作者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://www.ptt.cc/bbs/Python/M.1611496151.A.C91.html
推 fghdfh: 不要處理太多，就ok了 01/24 22:39
→ yfly823: 好 我再試試看 01/24 23:22
推 fishinair: 圖片大小? 01/25 15:11
→ DLHZ: 這些設定你直接去看code不就知道有沒有用了嗎 01/26 13:42
推 fghdfh: 補一句8G顯存不算多啊 01/26 16:00
推 remia81: 如果是608x608 8G VRAM會爆 01/26 21:27
推 oToToT: 自己覺得8G少到不行 02/01 03:36
→ mureka: 直接去gcp租一個TeslaV100，並且客製2cores-8GB RAM,一個 02/07 13:30
→ mureka: 月兩千多。蠻划算的 02/07 13:30