为什么我运营yolov3训练的时候一直内存溢出
收藏
用的飞桨上几个常见的yolov3项目,每次都是到训练阶段开始内存溢出,用了高级算力也是这样。连demo都跑不了,该怎么解决呢?
0
收藏
请登录后评论
fork的原始项目链接是什么?>
是不是你的batchsize每批次送入的数据太大了呀
有遇到过相同的问题
解决方案:
1.重启notebook(如果不行,就试方法2)
2.设置使用的GPU号(还不行,试试方法3)
3.减小batch_size
应该是读取数据的问题,之前我也遇到过
可以修改yolov3.py里面数据部分,里面有个buffsize,设置的太大导致的
buffsize太大导致数据一直在加载到内存,最后内存溢出
这种情况往往可以迭代几次
之前我也是这么解决的
或者把具体的错误贴一下?
bs设置太大了吧
一般情况 一直训练重启就释放了 第一次就炸应该是bs的问题
不一定 bufsize也有影响
明白了
试下pp-yolo哈哈哈
听闻很强
比yolov4还要猛
对对对,还没试过
AI Studio就有项目,可以去看看
好的 这就去看看