ColossalAI/tests/test_booster
Frank Lee 1beb85cc25
[checkpoint] refactored the API and added safetensors support (#3427)
* [checkpoint] refactored the API and added safetensors support

* polish code
2023-04-04 15:23:01 +08:00
..
test_mixed_precision [booster] added the plugin base and torch ddp plugin (#3180) 2023-03-21 17:39:30 +08:00
test_plugin [checkpoint] refactored the API and added safetensors support (#3427) 2023-04-04 15:23:01 +08:00
test_accelerator.py [booster] added the plugin base and torch ddp plugin (#3180) 2023-03-21 17:39:30 +08:00