github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/zero
History
Jiarui Fang 7675366fce
[polish] rename col_attr -> colo_attr (#558) 		2022-03-31 12:25:45 +08:00
..
init_ctx [polish] rename col_attr -> colo_attr (#558) 2022-03-31 12:25:45 +08:00
shard_utils [zero] non model data tracing (#545) 2022-03-29 15:45:48 +08:00
sharded_model [polish] rename col_attr -> colo_attr (#558) 2022-03-31 12:25:45 +08:00
sharded_optim [polish] rename col_attr -> colo_attr (#558) 2022-03-31 12:25:45 +08:00
sharded_param [zero] hijack p.grad in sharded model (#554) 2022-03-30 18:14:50 +08:00
__init__.py [refactor] remove old zero code (#517) 2022-03-25 14:54:39 +08:00