[Refactor] Replace torch distributed with mmengine dist module (#196)
* [Fix] Replace torch distributed with mmengine dist module * minor refinement * move all_reduce_params to dist.py * add unit tests * update unit tests * fix test_logger.py * add examples
Showing
- mmengine/dist/__init__.py 3 additions, 3 deletionsmmengine/dist/__init__.py
- mmengine/dist/dist.py 94 additions, 2 deletionsmmengine/dist/dist.py
- mmengine/hooks/sync_buffer_hook.py 3 additions, 78 deletionsmmengine/hooks/sync_buffer_hook.py
- mmengine/logging/logger.py 3 additions, 5 deletionsmmengine/logging/logger.py
- mmengine/runner/runner.py 0 additions, 3 deletionsmmengine/runner/runner.py
- tests/test_dist/test_dist.py 83 additions, 0 deletionstests/test_dist/test_dist.py
- tests/test_logging/test_logger.py 2 additions, 6 deletionstests/test_logging/test_logger.py
Loading
Please register or sign in to comment