ColossalAI/colossalai/zero/legacy
digger yu a9d1cadc49
fix typo with colossalai/trainer utils zero (#3908)
2023-06-07 16:08:37 +08:00
..
gemini fix typo with colossalai/trainer utils zero (#3908) 2023-06-07 16:08:37 +08:00
init_ctx [bf16] add bf16 support (#3882) 2023-06-05 15:58:31 +08:00
shard_utils
sharded_model
sharded_optim
sharded_param
__init__.py