作者yfly823 (欢乐打撞球~)
看板Python
标题[问题] Yolo V3 Pytorch - memory issue
时间Sun Jan 24 21:49:09 2021
最近在练习用 Yolo V3 Trainning，
但是一直遇到 Out memory 的问题 (用的是 RTX2070 显卡)
我已经查了蛮多方法但是都没有很好的效果
依照网路查到的去改变 Cfg 参数根本没有影响，
像是 Cfg 里的 Batch Size 或 subdivisions都没有改变
只有在下 Comment Train.py 作training 时直接改变 --Batch Size 到 <6 训练,才可
能可以 Training
不知道有没有强者知道我的状况是甚麽呢 ?
因为我只是先试跑一下大概跑50-60张图片而已
这样真的就可以把 8G 显存吃掉吗 ?
另外，想问一个笨笨的问题
我在下Training comment 里若直接指定 -- Batch Size 大小
是否Cfg 档里的 Batch Size 设定多少都没有关系了呢 ?
以上，希望有前辈能帮忙解惑了!!
谢谢
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 1.162.94.229 (台湾)
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1611496151.A.C91.html
1F：推 fghdfh: 不要处理太多，就ok了 01/24 22:39
2F：→ yfly823: 好 我再试试看 01/24 23:22
3F：推 fishinair: 图片大小? 01/25 15:11
4F：→ DLHZ: 这些设定你直接去看code不就知道有没有用了吗 01/26 13:42
5F：推 fghdfh: 补一句8G显存不算多啊 01/26 16:00
6F：推 remia81: 如果是608x608 8G VRAM会爆 01/26 21:27
7F：推 oToToT: 自己觉得8G少到不行 02/01 03:36
8F：→ mureka: 直接去gcp租一个TeslaV100，并且客制2cores-8GB RAM,一个 02/07 13:30
9F：→ mureka: 月两千多。蛮划算的 02/07 13:30