mirror of https://github.com/hpcaitech/ColossalAI
d26902645e
* add save/load state dict for ColoDDP * add unit test * refactor unit test folder * polish unit test * rename unit test |
||
---|---|---|
.. | ||
test_checkpoint | ||
test_activation_checkpointing.py | ||
test_commons.py | ||
test_lazy_init_ctx.py | ||
test_memory.py | ||
test_zero_gradient_clippling.py |