1. 程式人生 > >yolov3 訓練coco資料(cuda8.0-gxt1080)

yolov3 訓練coco資料(cuda8.0-gxt1080)

1. train遇到這個問題:
已放棄Resizing
608
CUDA Error: out of memory
darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.

百度的說法是gpu架構和作者預設使用的不一致。所以根據自己的gpu型號:gxt1080(pascal支援,百度即可得知...)。然後上官網可以查到gpu對應的comput_*:


但是呢我把這個修改好了,再train的時候,還是出現resizing * 出現 out of memory.....


難道是gpu的視訊記憶體不夠?....   或許吧.....

1:CUDA Error: out of memory表明視訊記憶體不夠,調小batch數值

2:Resizing到不同尺度,同樣增加視訊記憶體。設定random=0。關閉多尺度訓練。