Skip to content
Snippets Groups Projects
Unverified Commit 0857f9fb authored by Hakjin Lee's avatar Hakjin Lee Committed by GitHub
Browse files

[Feature] Support torch ZeroRedundancyOptimizer (#551)


* [Feature] Support torch ZeRORedundancyOptimizer

Co-authored-by: default avatarJunhwa Song <ethan9867@gmail.com>
Signed-off-by: default avatarJunhwa Song <ethan9867@gmail.com>
Signed-off-by: default avatarHakjin 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: default avatarZaida 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: default avatarJunhwa Song <ethan9867@gmail.com>
Signed-off-by: default avatarHakjin Lee <nijkah@gmail.com>
Co-authored-by: default avatarJunhwa Song <ethan9867@gmail.com>
Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
parent bf369da2
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment