在Google Colab上运行Darknet YOLOv4时CUDA错误



我认为Colab或Cuda最近的更新抛弃了YOLOv4模型。我以前建立了这个笔记本电脑,并训练了一个完整的模型,没有问题,但现在当我运行完全相同的代码,没有任何变化,我得到这个问题:

672 x 672 
try to allocate additional workspace_size = 65.03 MB 
CUDA allocate done! 
Loaded: 2.962808 seconds - performance bottleneck on CPU or Disk HDD/SSD
CUDA Error Prev: an illegal memory access was encountered
Darknet error location: ./src/dark_cuda.c, check_error, line #81
CUDA Error Prev: an illegal memory access was encountered: File exists

还有其他人有这个问题吗?这似乎是最近才发生的,因为到目前为止还没有人能弄清楚。我是这样运行我的暗网的:

!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg yolov4.conv.137 -dont_show -map -clear

看来Google Colab已经减少了免费笔记本的内存。

尝试在配置中更改批处理大小和细分。

batch=32 and subdivision =8为我解决了这个问题。

当我将图像大小从608x608更改为416x416时,我也面临同样的问题。之后它就可以工作了

最新更新