ColossalAI/tests/test_utils/test_checkpoint
FrankLeeeee 62b4ce7326 [test] added missing decorators to model checkpointing tests 2022-04-12 11:08:15 +08:00
..
test_checkpoint_1d.py [test] added missing decorators to model checkpointing tests 2022-04-12 11:08:15 +08:00
test_checkpoint_2d.py [test] added missing decorators to model checkpointing tests 2022-04-12 11:08:15 +08:00
test_checkpoint_2p5d.py [test] added missing decorators to model checkpointing tests 2022-04-12 11:08:15 +08:00
test_checkpoint_3d.py [test] added missing decorators to model checkpointing tests 2022-04-12 11:08:15 +08:00