YH a22407cc02 [zero] Suggests a minor change to confusing variable names in the ZeRO optimizer. (#3173) ...

* Fix confusing variable name in zero opt

* Apply lint

* Fix util func

* Fix minor util func

* Fix zero param optimizer name

2023-04-27 18:43:14 +08:00

..

__init__.py

[zero] reorganize zero/gemini folder structure (#3424)

2023-04-04 13:48:16 +08:00

base_store.py

[zero] reorganize zero/gemini folder structure (#3424)

2023-04-04 13:48:16 +08:00

bucket_store.py

[zero] reorganize zero/gemini folder structure (#3424)

2023-04-04 13:48:16 +08:00

gradient_store.py

[zero] reorganize zero/gemini folder structure (#3424)

2023-04-04 13:48:16 +08:00

parameter_store.py

[zero] Suggests a minor change to confusing variable names in the ZeRO optimizer. (#3173)

2023-04-27 18:43:14 +08:00

tensor_bucket.py

[zero] reorganize zero/gemini folder structure (#3424)

2023-04-04 13:48:16 +08:00