SSD TensorFlow在训练和测试过程中遇到Data loss error等问题
发布时间
阅读量:
阅读量
实验环境:
操作系统: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

还没有任何评论哟~
