[问题] Yolo V3 Pytorch - memory issue

楼主: yfly823 (欢乐打台球~)   2021-01-24 21:49:09
最近在练习用 Yolo V3 Trainning,
但是一直遇到 Out memory 的问题 (用的是 RTX2070 显卡)
我已经查了蛮多方法但是都没有很好的效果
依照网络查到的去改变 Cfg 参数根本没有影响,
像是 Cfg 里的 Batch Size 或 subdivisions都没有改变
只有在下 Comment Train.py 作training 时直接改变
作者: fghdfh (fghdfh)   2021-01-24 22:39:00
不要处理太多,就ok了
楼主: yfly823 (欢乐打台球~)   2021-01-24 23:22:00
好 我再试试看
作者: fishinair (经纪人)   2021-01-25 15:11:00
图片大小?
作者: DLHZ ( )   2021-01-26 13:42:00
这些设定你直接去看code不就知道有没有用了吗
作者: fghdfh (fghdfh)   2021-01-26 16:00:00
补一句8G显存不算多啊
作者: remia81   2021-01-26 21:27:00
如果是608x608 8G VRAM会爆
作者: oToToT (屁孩)   2021-02-01 03:36:00
自己觉得8G少到不行
作者: mureka (mureka)   2021-02-07 13:30:00
直接去gcp租一个TeslaV100,并且客制2cores-8GB RAM,一个月两千多。蛮划算的

Links booklink

Contact Us: admin [ a t ] ucptt.com