mirror of https://github.com/hpcaitech/ColossalAI
e5d615aeee
* remove hybrid adam in test_moe_zero_optim * fix activation checkpointing and its unitest |
||
---|---|---|
.. | ||
components_to_test | ||
test_amp | ||
test_comm | ||
test_config | ||
test_context | ||
test_data | ||
test_data_pipeline_tensor_parallel | ||
test_engine | ||
test_layers | ||
test_moe | ||
test_optimizer | ||
test_trainer | ||
test_utils | ||
test_zero_data_parallel | ||
__init__.py |