tensorflow object detection api 训练问题

来源:6-24 如何将训练好的模型转化成pb文件

reapp

2019-07-24

老师你好,我用tensorflow object detection api 训练的时候,发现GPU利用率很低,而且浮动很大(1%~50%) ,刚觉平均大概只有百分之十几。我的显卡是titan XP (12g显存基本用满了),由于GPU利用率很低,训练速度较慢。请问这是怎么回事呢(我只修改了训练的配置文件,没有改动源码)?

写回答

1回答

会写代码的好厨师

2019-07-25

你训练一晚上大概训练多少次?输入图多大?原始图片多大。一般这种情况是因为数据处理的部分耗费了很多时间。想办法节省这边的时间。先回答我上面几个问题,我们再来定位具体的原因。

0
1
reapp
一晚上跑了十个小时左右大概能跑三四万歩(batchsize=32),输入图像416x416,原始图像1280x720
2019-07-25
共1条回复

Python3+TensorFlow打造人脸识别智能小程序

理论与实战项目双管齐下,让AI技术真正落地应用,适合毕设展示。

1086 学习 · 538 问题

查看课程