ColossalAI/colossalai/zero/sharded_optim
YH 7b13f7db18
[zero] trivial zero optimizer refactoring (#2869)
* Fix mionr grad store interface

* Apply lint
2023-02-27 14:04:53 +08:00
..
bookkeeping [zero] trivial zero optimizer refactoring (#2869) 2023-02-27 14:04:53 +08:00
__init__.py [zero] migrate zero1&2 (#1878) 2022-11-11 09:26:40 +08:00
_utils.py [zero] fix wrong import (#2777) 2023-02-17 10:26:07 +08:00
low_level_optim.py [zero] trivial zero optimizer refactoring (#2869) 2023-02-27 14:04:53 +08:00
sharded_optim_v2.py fix move fp32 shards (#1604) 2022-09-16 17:33:16 +08:00