[Feature] Support model ema with EMAHook - Githubissues

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

https://detrex.readthedocs.io/en/latest/

Apache License 2.0

2.01k stars 209 forks source link

[Feature] Support model ema with EMAHook #201

Closed rentainhe closed 1 year ago

rentainhe commented 1 year ago

TODO

Support EMAHook as d2go.
Testing DINO-R50-4scale-12ep with ModelEMA: 49.4AP.

Note

If you set train.model_ema.enabled=True, after each eval_period, detrex will conduct two evalution on normal model and its ema version.

Usage

To enabled model ema training:

python tools/train_net.py --config-file projects/dino/configs/dino_r50_4scale_12ep.py \
                      --num-gpus 8 \
                      --resume \
                      train.model_ema.enabled=True \

Testing model with ema states when eval-only:

python tools/train_net.py --config-file projects/dino/configs/dino_r50_4scale_12ep.py \
                      --num-gpus 8 \
                      --resume \
                      train.model_ema.enabled=True \
                      train.model_ema.use_ema_weights_for_eval_only=True \