ashkamath / mdetr

Apache License 2.0
969 stars 125 forks source link

Why is `model_ema` not saved in `BEST_checkpoint.pth`? #67

Open TopCoder2K opened 2 years ago

TopCoder2K commented 2 years ago

Hi!

I'm a little confused, the evaluation is performed with model_ema and the metrics are calculated for this version of the model, so, it seems I should use model_ema for the inference. But why is model_without_ddp saved in BEST_checkpoint and model_ema is not?..