ColossalAI/colossalai/zero/sharded_param
Jiarui Fang 37cb70feec
[zero] get memory usage for sharded param (#536)
2022-03-28 15:01:21 +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 for sharded param (#536) 2022-03-28 15:01:21 +08:00
sharded_tensor.py [zero] refactor model data tracing (#522) 2022-03-25 18:03:32 +08:00