search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.2k stars 1.27k forks source link

Our train.py #437

Open mnslarcher opened 4 years ago

mnslarcher commented 4 years ago

Hi @zylo117,

If you want to take a look, I put on GitHub the train.py that my team and I are using, maybe you will find something you want to include in your train.py:

https://github.com/mnslarcher/Yet-Another-EfficientDet-Pytorch/blob/master/train.py

With the default hps, the train.py use the same configuration of the original paper, some hp is adapted automatically when another change (e.g. lr when batch size change, or l2 reg. when lr change, we took inspiration from here and here ).

This default configuration works well for our use case.

We use AWS SageMaker so some part is not 100% generic but it should be very easy to adapt it to any enviroment.

I take this opportunity to thank you for having made this package available, we are using it successfully to identify assets and anomalies in aerial photos.

zylo117 commented 4 years ago

Thank you so much for sharing your work too. It's also been a great inspiration for me.

mnslarcher commented 4 years ago

Thanks @zylo117 :)