2022年2月11日
摘要: 1、训练时出现:could not successfully open dynamic XXX.dll,导致不能使用GPU训练,而使用CPU训练 因为RTX3060 对应 CUDA11.1 对应 tensorflow 2.4 ,而自带的动态库是cusolver64_11.dll,需要cusolver 阅读全文
posted @ 2022-02-11 11:26 WenJXUST 阅读(58) 评论(0) 推荐(0) 编辑
摘要: 1、to_float,to_int32 # has_min = tf.to_float(tf.reduce_any(pos_num_neg_mask)) has_min = tf.cast(tf.reduce_any(pos_num_neg_mask), tf.float32) 2、tf.log # 阅读全文
posted @ 2022-02-11 11:15 WenJXUST 阅读(233) 评论(0) 推荐(0) 编辑