search

quangnhat185 / darknet_for_colab

Darknet repository for training YOLOv4 on Google Colab with Custom Dataset
MIT License
31 stars 57 forks source link

darknet: ./src/utils.c:326: error: Assertion `0' failed. #13

Closed HariTarz closed 1 year ago

HariTarz commented 1 year ago

While started training !./darknet detector train data/yolov4.data cfg/yolov4_custom_train.cfg yolov4.conv.137 -dont_show -map I got this error.

CUDA status Error: file: ./src/dark_cuda.c : () : line: 454 : build time: Oct 11 2022 - 11:14:30
CUDA Error: invalid argument
CUDA Error: invalid argument: Success
darknet: ./src/utils.c:326: error: Assertion '0' failed.

Any thoughts on why this is happening? It works only if the CUDNN_HALF=1, but the default configuration is CUDNN_HALF=0 in the Makefile.

quangnhat185 commented 1 year ago

This happens due to the upgrade CUDA on Colab. MAKEFILE has been updated to fix this issue