github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-07 02:54:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
feature/elixir
ColossalAI/tests/test_checkpoint_io
History
wukong1992 6b305a99d6
[booster] torch fsdp fix ckpt (#3788)
2023-05-23 16:58:45 +08:00
..
test_gemini_checkpoint_io.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00
test_general_checkpoint_io.py [booster] add tests for ddp and low level zero's checkpointio (#3715) 2023-05-10 12:17:02 +08:00
test_low_level_zero_checkpoint_io.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00
test_torch_ddp_checkpoint_io.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00
test_torch_fsdp_checkpoint_io.py [booster] torch fsdp fix ckpt (#3788) 2023-05-23 16:58:45 +08:00
utils.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2023-05-19 19:42:31 +08:00