ColossalAI/tests/test_booster/test_plugin
flybird11111 2a2ec49aa7
[plugin]fix 3d checkpoint load when booster boost without optimizer. (#5135)
* fix 3d checkpoint load when booster boost without optimizer

fix 3d checkpoint load when booster boost without optimizer

* test ci

* revert ci

* fix

fix
2023-11-30 18:37:47 +08:00
..
test_3d_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_dp_plugin_base.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_gemini_plugin.py [plugin]fix 3d checkpoint load when booster boost without optimizer. (#5135) 2023-11-30 18:37:47 +08:00
test_low_level_zero_plugin.py [npu] add npu support for gemini and zero (#5067) 2023-11-20 16:12:41 +08:00
test_torch_ddp_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_torch_fsdp_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00