You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_zero/test_legacy
Hongxin Liu ae02d4e4f7
[bf16] add bf16 support (#3882)
2 years ago
..
common.py [test] reorganize zero/gemini tests (#3445) 2 years ago
test_found_inf.py [test] refactor tests with spawn (#3452) 2 years ago
test_gemini_manager.py [test] refactor tests with spawn (#3452) 2 years ago
test_init_context.py [test] refactor tests with spawn (#3452) 2 years ago
test_param_op.py [test] refactor tests with spawn (#3452) 2 years ago
test_shard_model_v2.py [test] refactor tests with spawn (#3452) 2 years ago
test_shard_param.py [test] refactor tests with spawn (#3452) 2 years ago
test_sharded_optim_state_dict.py [test] refactor tests with spawn (#3452) 2 years ago
test_sharded_optim_v2.py [test] refactor tests with spawn (#3452) 2 years ago
test_sharded_optim_with_sync_bn.py [CI] Update test_sharded_optim_with_sync_bn.py (#3688) 2 years ago
test_state_dict.py [test] refactor tests with spawn (#3452) 2 years ago
test_tensor_utils.py [test] refactor tests with spawn (#3452) 2 years ago
test_zero_engine.py [bf16] add bf16 support (#3882) 2 years ago