RMSProp, adadelta and Adam optimizers with Yolo models

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )

http://pjreddie.com/darknet/

Other

21.75k stars 7.96k forks source link

RMSProp, adadelta and Adam optimizers with Yolo models #6907

Open hsaleem1 opened 4 years ago

hsaleem1 commented 4 years ago

Hi there,

Can anybody please guide me that is it possible to use Adam, RMSProp, Adadelta and other optimizers with Yolo models?

Thanks.

blake-espeland commented 3 years ago

I know for sure that the Adam optimizer is available, however, I'm not sure about the other two. You can set adam=1 in your [net] section to enable it. I'd recommend using learning_rate=0.001 with it.