[Feature] Support torch ZeroRedundancyOptimizer (#551)
* [Feature] Support torch ZeRORedundancyOptimizer Co-authored-by:Junhwa Song <ethan9867@gmail.com> Signed-off-by:
Junhwa Song <ethan9867@gmail.com> Signed-off-by:
Hakjin Lee <nijkah@gmail.com> * lint * Fix saving optimizer state_dict * Fix handling import error * Add test case * fix UT * Revert "fix UT" This reverts commit dd64538960ff7440c6020f533d43945ffc23f2d2. * fix handling import in UT * Fix saving zero checkpoint and delete redundant master_only * lint * test unittest * Fix handling impor error * Fix UT condition * Edit docstrings * Fix typo * Skip redundant procudure in checkpoint hook * fix typo again * Update mmengine/optim/optimizer/zero_optimizer.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Add api info * lint * Fix lint * Handling AmpOptimWrapper case * handling overlap_with_ddp * Fix error Signed-off-by:
Junhwa Song <ethan9867@gmail.com> Signed-off-by:
Hakjin Lee <nijkah@gmail.com> Co-authored-by:
Junhwa Song <ethan9867@gmail.com> Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Showing
- docs/en/api/optim.rst 1 addition, 0 deletionsdocs/en/api/optim.rst
- docs/zh_cn/api/optim.rst 1 addition, 0 deletionsdocs/zh_cn/api/optim.rst
- mmengine/hooks/checkpoint_hook.py 6 additions, 3 deletionsmmengine/hooks/checkpoint_hook.py
- mmengine/optim/optimizer/__init__.py 2 additions, 1 deletionmmengine/optim/optimizer/__init__.py
- mmengine/optim/optimizer/zero_optimizer.py 66 additions, 0 deletionsmmengine/optim/optimizer/zero_optimizer.py
- tests/test_optim/test_optimizer/test_optimizer.py 80 additions, 1 deletiontests/test_optim/test_optimizer/test_optimizer.py
Loading
Please register or sign in to comment