ColossalAI/colossalai/zero/sharded_param
Jiarui Fang a590ed0ba3
[zero] improve the accuracy of get_memory_usage of sharded param (#538)
2022-03-28 16:19:19 +08:00
..
__init__.py [zero] polish sharded param name (#484) 2022-03-22 14:36:16 +08:00
sharded_param.py [zero] improve the accuracy of get_memory_usage of sharded param (#538) 2022-03-28 16:19:19 +08:00
sharded_tensor.py [zero] refactor model data tracing (#522) 2022-03-25 18:03:32 +08:00