search

hustvl / YOLOS

[NeurIPS 2021] You Only Look at One Sequence
https://arxiv.org/abs/2106.00666
MIT License
827 stars 118 forks source link

AMP Support? #18

Closed impiga closed 2 years ago

impiga commented 2 years ago

Thanks for your great work and releasing the code!

I find that in engine.py, AMP-related code is commented out. And I am wondering that if I can use AMP in this project. Would it speed up the training and would it hurt the performance?

Yuxin-CV commented 2 years ago

Hi@impiga, thanks for your interests in our work!

We tried torch naive amp, and we observed no significant speed up and acc drop. Please also see https://github.com/facebookresearch/detr/issues/127.

impiga commented 2 years ago

Thanks for your response!