ColossalAI/tests/test_utils
HELSON e5d615aeee
[hotfix] fix bugs in testing (#659)
* remove hybrid adam in test_moe_zero_optim

* fix activation checkpointing and its unitest
2022-04-02 21:58:47 +08:00
..
test_checkpoint [model checkpoint] added unit tests for checkpoint save/load (#599) 2022-04-01 16:53:32 +08:00
test_activation_checkpointing.py [hotfix] fix bugs in testing (#659) 2022-04-02 21:58:47 +08:00
test_bucket_tensor_copy.py [memory] set cuda mem frac (#506) 2022-03-24 16:57:13 +08:00
test_commons.py [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
test_gradient_accumluation.py [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00
test_tensor_move.py [zero] test zero tensor utils (#609) 2022-04-01 15:16:59 +08:00
test_zero_gradient_clippling.py [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00