ColossalAI/tests/test_checkpoint_io
Baizhou Zhang 0bb0b481b4 [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
..
test_gemini_checkpoint_io.py [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
test_general_checkpoint_io.py
test_low_level_zero_checkpoint_io.py
test_torch_ddp_checkpoint_io.py
test_torch_fsdp_checkpoint_io.py
utils.py