暗网： ./src/utils.c：325： 错误： 断言 '0' 失败 (YOLO-v4) 谷歌合作

Question

我试图使用Google Colab在自己的数据集上训练YOLO_v4，但遇到了下一个错误：

CUDA status Error: file: ./src/blas_kernels.cu : () : line: 841 : build time: Oct  9 2020 - 12:04:13 
CUDA Error: no kernel image is available for execution on the device
CUDA Error: no kernel image is available for execution on the device: File exists
darknet: ./src/utils.c:325: error: Assertion `0' failed.
CPU times: user 58 ms, sys: 12.4 ms, total: 70.4 ms
Wall time: 15.4 s

我已经检查了我的代码，但没有发现错误。然后我用我使用的例子运行了谷歌Colab笔记本(https://colab.research.google.com/drive/1cbzkKcACQPzLXuBYH7KPOh73Ko8xdUPn#scrollTo=6miYFbvExqMd)。当我执行单元格时：

%%time
!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg -dont_show -map #./backup/custom-yolov4-detector_last.weights
#If you get CUDA out of memory adjust subdivisions above!
#adjust max batches down for shorter training above

这个错误有时出现并没有。我很困惑。。。我认为这在某种程度上与我连接的GPU的可用性或类型有关。有什么建议吗？

Answer 1

检查此链接并下载暗网，如这里所示在这里输入链接描述

然后创建一个yolov4.conf文件，并根据类标签的数量对其进行配置。你可以在这里找到这个配置的例子：在这里输入链接描述