1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-04 14:38:10 +00:00
Commit Graph

7 Commits

Author SHA1 Message Date
Hongxin Liu
6552cbf8e1
[booster] fix no_sync method ()
* [booster] fix no_sync method

* [booster] add test for ddp no_sync

* [booster] fix merge

* [booster] update unit test

* [booster] update unit test

* [booster] update unit test
2023-05-09 11:10:02 +08:00
Hongxin Liu
3bf09efe74
[booster] update prepare dataloader method for plugin ()
* [booster] add prepare dataloader method for plug

* [booster] update examples and docstr
2023-05-08 15:44:03 +08:00
Hongxin Liu
d0915f54f4
[booster] refactor all dp fashion plugins ()
* [booster] add dp plugin base

* [booster] inherit dp plugin base

* [booster] refactor unit tests
2023-05-05 19:36:10 +08:00
Frank Lee
7d8d825681
[booster] fixed the torch ddp plugin with the new checkpoint api () 2023-04-06 09:43:51 +08:00
Frank Lee
1beb85cc25
[checkpoint] refactored the API and added safetensors support ()
* [checkpoint] refactored the API and added safetensors support

* polish code
2023-04-04 15:23:01 +08:00
Frank Lee
73d3e4d309
[booster] implemented the torch ddd + resnet example ()
* [booster] implemented the torch ddd + resnet example

* polish code
2023-03-27 10:24:14 +08:00
Frank Lee
e7f3bed2d3
[booster] added the plugin base and torch ddp plugin ()
* [booster] added the plugin base and torch ddp plugin

* polish code

* polish code

* polish code
2023-03-21 17:39:30 +08:00