search

IDEA-Research / DN-DETR

[CVPR 2022 Oral] Official implementation of DN-DETR
Apache License 2.0
541 stars 62 forks source link

optimizer zero_grad before step #18

Closed K-Mike closed 2 years ago

K-Mike commented 2 years ago

https://github.com/IDEA-opensource/DN-DETR/blob/206fa267ba7df978fa968edda9f7dd351a4b72c1/engine.py#L76

It looks like that you should call zero_grad after scaler.update()