mirror of https://github.com/hpcaitech/ColossalAI
e5d615aeee
* remove hybrid adam in test_moe_zero_optim * fix activation checkpointing and its unitest |
||
---|---|---|
.. | ||
test_checkpoint | ||
test_activation_checkpointing.py | ||
test_bucket_tensor_copy.py | ||
test_commons.py | ||
test_gradient_accumluation.py | ||
test_tensor_move.py | ||
test_zero_gradient_clippling.py |