[問題] Yolo V3 Pytorch - memory issue

看板Python作者yfly823 (歡樂打撞球~)時間3年前 (2021/01/24 21:49)推噓4(4推 0噓 2→)

留言6則, 5人參與討論串1/1

最近在練習用 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

01/24 22:39, 3年前 , 1^F

01/24 22:39, 1^F

→

yfly823

01/24 23:22, 3年前 , 2^F

01/24 23:22, 2^F

推

fishinair

01/25 15:11, 3年前 , 3^F

01/25 15:11, 3^F