[Feature] Add test time augmentation base model. (#538)
* First commit * add BaseTestTimeAugModel * Add unit test * move loop logic to test_step * fix ddp test * rename model to module * optim __init__ * Fix as comment * Fix as comment * make val_step should not be called * make tta do not inherit base model * Fix unit test * Enhance docstring * Fix as comment * Fix as comment * minor refine * minor refine * minor refine * fix unit test * minor refine * minor refine * minor refine * minor refine * minor refine * minor refine * fix unit test
Showing
- docs/en/api/model.rst 1 addition, 0 deletionsdocs/en/api/model.rst
- docs/zh_cn/api/model.rst 1 addition, 0 deletionsdocs/zh_cn/api/model.rst
- mmengine/model/__init__.py 2 additions, 2 deletionsmmengine/model/__init__.py
- mmengine/model/wrappers/__init__.py 2 additions, 1 deletionmmengine/model/wrappers/__init__.py
- mmengine/model/wrappers/distributed.py 2 additions, 4 deletionsmmengine/model/wrappers/distributed.py
- mmengine/model/wrappers/test_time_aug.py 126 additions, 0 deletionsmmengine/model/wrappers/test_time_aug.py
- mmengine/runner/checkpoint.py 2 additions, 2 deletionsmmengine/runner/checkpoint.py
- mmengine/runner/runner.py 29 additions, 11 deletionsmmengine/runner/runner.py
- tests/test_hooks/test_ema_hook.py 7 additions, 7 deletionstests/test_hooks/test_ema_hook.py
- tests/test_model/test_wrappers/test_test_aug_time.py 55 additions, 0 deletionstests/test_model/test_wrappers/test_test_aug_time.py
- tests/test_runner/test_runner.py 29 additions, 0 deletionstests/test_runner/test_runner.py
Loading
Please register or sign in to comment