github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-18 08:35:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
73d36618a6263b3e0ff3178a4e7dfaaec60655b0
ColossalAI/colossalai/zero/sharded_optim
History
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
Feature/zero (#279)
2022-03-11 15:50:28 +08:00
sharded_optim_v2.py
[zero] optimize grad offload (#539)
2022-03-29 12:48:00 +08:00