johnolafenwa / deepstack-trainer

Custom Object Detection Training for DeepStack
GNU General Public License v3.0
20 stars 12 forks source link

Lower epochs by default? #4

Open robmarkcole opened 3 years ago

robmarkcole commented 3 years ago

300 appears overkill if this is correct, although I have only a single class.

image
robmarkcole commented 3 years ago

On looking at my old notes, 100 epoch looks like a sensible value

https://github.com/robmarkcole/fire-detection-from-images/tree/master/pytorch/object-detection/yolov5