ColossalAI/colossalai/zero/sharded_param
ver217 36f9a74ab2 fix sharded param hook and unit test 2022-03-11 15:50:28 +08:00
..
__init__.py [zero] sharded tensor (#305) 2022-03-11 15:50:28 +08:00
sharded_param.py fix sharded param hook and unit test 2022-03-11 15:50:28 +08:00
sharded_tensor.py [zero] sharded tensor (#305) 2022-03-11 15:50:28 +08:00