ColossalAI/colossalai/zero/sharded_optim
ver217 fb841dd5c5
[zero] optimize grad offload (#539)
* optimize grad offload

* polish code

* polish code
2022-03-29 12:48:00 +08:00
..
__init__.py [refactor] remove old zero code (#517) 2022-03-25 14:54:39 +08:00
_utils.py
sharded_optim_v2.py [zero] optimize grad offload (#539) 2022-03-29 12:48:00 +08:00