search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.8k stars 7.97k forks source link

how to use cutmix=1? #5278

Open Tongyuang opened 4 years ago

Tongyuang commented 4 years ago

I was training on my own dataset. The backbone was yolov3-tiny. I added cutmix=1 on the [net] layer, but after I run the command

./darknet detector train ... The training process just stopped after the sentence Done! Loaded xx layers from weights-file was printed on the screen. No error If I remove the command cutmix=1, the trainig process can run normally. I checked the gpu and the memory is sufficient. I wonder if I didn't edit the .cfg file correctly :( image image

lsd1994 commented 4 years ago

Cutmix is only for classifier #4419