[問題] 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