github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/zero/sharded_param
History
Jiarui Fang c11ff81b15
[zero] get memory usage of sharded optim v2. (#542) 		2022-03-29 09:08:18 +08:00
..
__init__.py [zero] polish sharded param name (#484) 2022-03-22 14:36:16 +08:00
sharded_param.py [zero] get memory usage of sharded optim v2. (#542) 2022-03-29 09:08:18 +08:00
sharded_tensor.py [zero] refactor model data tracing (#522) 2022-03-25 18:03:32 +08:00