[Feature]: add base model, ddp model wrapper and unit test (#268)
* add base model, ddp model and unit test * add unit test * fix unit test * fix docstring * fix cpu unit test * refine base data preprocessor * refine base data preprocessor * refine interface of ddp module * remove optimizer hook * add forward * fix as comment * fix unit test * fix as comment * fix build optimizer wrapper * rebase main and fix unit test * stack_batch support stacking ndim tensor, add docstring for merge dict * fix lint * fix test loop * make precision_context effective to data_preprocessor * fix as comment * fix as comment * refine docstring * change collate_data output typehints * rename to_rgb to bgr_to_rgb and rgb_to_bgr * support build basemodel with built DataPreprocessor * fix as comment * fix docstring
Showing
- mmengine/hooks/__init__.py 2 additions, 3 deletionsmmengine/hooks/__init__.py
- mmengine/hooks/optimizer_hook.py 0 additions, 129 deletionsmmengine/hooks/optimizer_hook.py
- mmengine/hooks/runtime_info_hook.py 1 addition, 1 deletionmmengine/hooks/runtime_info_hook.py
- mmengine/model/__init__.py 10 additions, 5 deletionsmmengine/model/__init__.py
- mmengine/model/base_model/__init__.py 9 additions, 0 deletionsmmengine/model/base_model/__init__.py
- mmengine/model/base_model/base_model.py 256 additions, 0 deletionsmmengine/model/base_model/base_model.py
- mmengine/model/base_model/data_preprocessor.py 213 additions, 0 deletionsmmengine/model/base_model/data_preprocessor.py
- mmengine/model/base_module.py 1 addition, 1 deletionmmengine/model/base_module.py
- mmengine/model/utils.py 131 additions, 0 deletionsmmengine/model/utils.py
- mmengine/model/wrappers/__init__.py 6 additions, 2 deletionsmmengine/model/wrappers/__init__.py
- mmengine/model/wrappers/data_parallel.py 0 additions, 149 deletionsmmengine/model/wrappers/data_parallel.py
- mmengine/model/wrappers/distributed.py 123 additions, 0 deletionsmmengine/model/wrappers/distributed.py
- mmengine/model/wrappers/seperate_distributed.py 124 additions, 0 deletionsmmengine/model/wrappers/seperate_distributed.py
- mmengine/runner/loops.py 21 additions, 11 deletionsmmengine/runner/loops.py
- mmengine/runner/runner.py 41 additions, 33 deletionsmmengine/runner/runner.py
- mmengine/utils/misc.py 2 additions, 3 deletionsmmengine/utils/misc.py
- tests/test_hook/test_ema_hook.py 20 additions, 17 deletionstests/test_hook/test_ema_hook.py
- tests/test_hook/test_optimizer_hook.py 0 additions, 115 deletionstests/test_hook/test_optimizer_hook.py
- tests/test_hook/test_runtime_info_hook.py 1 addition, 6 deletionstests/test_hook/test_runtime_info_hook.py
- tests/test_logging/test_message_hub.py 1 addition, 1 deletiontests/test_logging/test_message_hub.py
Loading
Please register or sign in to comment