mirrors
/
mmengine
mirror of https://github.com/open-mmlab/mmengine.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmengine/docs/zh_cn/api
History
Hakjin Lee 0857f9fb40
[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 dd64538960.

* 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>
 		2022-10-27 20:31:50 +08:00
..
config.rst [Docs] Make API of docs clearer (#469) 2022-08-30 16:27:20 +08:00
dataset.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
device.rst [Feature] Support engine with NPU backend. (#572) 2022-10-24 11:42:29 +08:00
dist.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
evaluator.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
fileio.rst [Refactor] Refactor fileio without breaking back compatibility (#533) 2022-09-26 14:30:40 +08:00
hooks.rst [Docs] Make API of docs clearer (#469) 2022-08-30 16:27:20 +08:00
hub.rst [Docs] Fix API files of en docs (#525) 2022-09-13 14:57:18 +08:00
logging.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
model.rst [Feature] Add test time augmentation base model. (#538) 2022-10-10 14:29:33 +08:00
optim.rst [Feature] Support torch ZeroRedundancyOptimizer (#551) 2022-10-27 20:31:50 +08:00
registry.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
runner.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
structures.rst [Docs] Make API of docs clearer (#469) 2022-08-30 16:27:20 +08:00
utils.dl_utils.rst [Docs] Refactor docs structure (#494) 2022-08-31 21:18:55 +08:00
utils.rst [Feature] Add a function to mark the deprecated function. (#609) 2022-10-24 14:49:48 +08:00
visualization.rst [Docs] Make API of docs clearer (#469) 2022-08-30 16:27:20 +08:00