ColossalAI/colossalai/zero/sharded_param
Jiarui Fang 8d8c5407c0
[zero] refactor model data tracing (#522)
2022-03-25 18:03:32 +08:00
..
__init__.py [zero] polish sharded param name (#484) 2022-03-22 14:36:16 +08:00
sharded_param.py [zero] sharded model support the reuse of fp16 shard (#495) 2022-03-23 14:59:59 +08:00
sharded_tensor.py [zero] refactor model data tracing (#522) 2022-03-25 18:03:32 +08:00