Advertisement

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

![在这里插入图片描述](https://ad.itadn.com/c/weblog/blog-img/images/20

全部评论 (0)

还没有任何评论哟~