mirror of https://github.com/hpcaitech/ColossalAI
50ec3a7e06
* [test] skip tests when not enough GPUs are detected * polish code * polish code |
||
---|---|---|
.. | ||
test_checkpoint | ||
test_activation_checkpointing.py | ||
test_commons.py | ||
test_memory.py | ||
test_pipelinable.py | ||
test_zero_gradient_clippling.py |