在Ubuntu 18.04环境下，使用TensorFlow 1.14和Python3.6进行SSD模型训练时，遇到了DatalossError等问题。通过调整GPU配置和代码修改解决了问题。测试阶段，分别针对variables.Variable转换、eval_ssd_network.py中flatten操作进行了修正，确保了模型的正常运行。
环境：
Ubuntu 18.04
Tensorflow 1.14
Python3.6
Cuda10.0
Cudnn7.4.2
下载的ssd_tensorflow进行实验、训练，期间出现了一些问题，记录如下：
训练遇到的问题：
方法：将
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction)
config = tf.ConfigProto(log_device_placement=False, gpu_options=gpu_options)
修改为config = tf.ConfigProto() config.gpu_options.allow_growth = True
测试时遇到的问题：
1
将return variables.Variable 改为 return variables.VariableV1
2
在eval_ssd_network.py里面添加如下函数：
def flatten(x):
最低0.47元/天 解锁文章
扫一扫
1411
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
