[Feature] Support convert `BN` to `SyncBN` by config (#506)
* [Feature] Support convert BN to SyncBN by config * make unit test compatible with cpu * Fix as comment * fix unit test * change signature of convert_sync_batchnorm: rename sync_bn to implemention * fix unit test * 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 4 additions, 3 deletionsmmengine/model/__init__.py
- mmengine/model/utils.py 48 additions, 0 deletionsmmengine/model/utils.py
- mmengine/runner/runner.py 12 additions, 3 deletionsmmengine/runner/runner.py
- tests/test_model/test_model_utils.py 26 additions, 1 deletiontests/test_model/test_model_utils.py
- tests/test_runner/test_runner.py 32 additions, 1 deletiontests/test_runner/test_runner.py
Loading
Please register or sign in to comment