Open only-zzx opened 3 years ago
reduce learning rate
reduce learning rate
my learning_rate=0.0001. have to reduce more?
I don't know your dataset and this model in detail, you can just try it. I got the similar chart with you and just solved by reducing learning rate.
Has your problem been solved,I have the same problem as you。
have an issue with Nan avg-loss: while training my own dateset in p5,i found avg-loss getting bigger and bingger. but same dataset has good result in YOLOv4 traning. I chang the cfg parameter "batch=10" to fix my GPU cfg-file is here:yolov4-p5.txt
i use command:./darknet detector train cfg/voc.data cfg/yolov4-p5.cfg ./yolov4-p5.conv.232 -gpus 0,1,2,3 -map my