ColossalAI/tests/test_utils/test_checkpoint
Frank Lee 2b2dc1c86b
[pipeline] refactor the pipeline module (#1087)
* [pipeline] refactor the pipeline module

* polish code
2022-06-10 11:27:38 +08:00
..
test_checkpoint_1d.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_checkpoint_2d.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_checkpoint_2p5d.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_checkpoint_3d.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00