AMP Support?

[Yutong Lin]

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 Fang (方羽新)]

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 facebookresearch/detr#127.

[Yutong Lin]

Thanks for your response!