ColossalAI/tests/test_zero_data_parallel
HELSON 055fbf5be6
[zero] adapt zero for unsharded paramters (Optimizer part) (#601)
2022-04-01 20:10:47 +08:00
..
common.py [zero] adapt zero for unsharded paramters (Optimizer part) (#601) 2022-04-01 20:10:47 +08:00
test_init_context.py [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
test_shard_model_v2.py
test_shard_param.py
test_sharded_optim_v2.py
test_sharded_optim_with_sync_bn.py
test_state_dict.py
test_zero_engine.py