我认为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时,我也面临同样的问题。之后它就可以工作了