darknet icon indicating copy to clipboard operation
darknet copied to clipboard

max_batches in yolov4-custom.cfg

Open TidelyPom opened this issue 1 year ago • 2 comments

Hi. I am training for custom objects of very different sizes in the image. As recommended I am using yolov4-custom.cfg instead of yolo-obj.cfg but there is something very different in yolov4-custom.cfg from yolo-obj.cfg : on lines 20 and 22 the values are 500500 for max_batches and 400000, 450000 for steps instead of 6000 and 4800, 5400. Is this the recommended values ? I have only 2 types of objects now and maybe 3 in the future. The problem with these values is that the training time is estimated at 450 hours, which is a bit long. Thank you

TidelyPom avatar Nov 16 '23 12:11 TidelyPom

You need to re-read the steps on training. See https://github.com/hank-ai/darknet#training

stephanecharette avatar Nov 16 '23 17:11 stephanecharette

Also of possible interest to you: https://www.ccoderun.ca/programming/yolo_faq/#configuration_template

stephanecharette avatar Nov 16 '23 17:11 stephanecharette