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 |
|
Frank Lee
|
638a07a7f9
|
[test] fixed gemini plugin test (#3411)
* [test] fixed gemini plugin test
* polish code
* polish code
|
2023-04-03 17:12:22 +08:00 |
|
ver217
|
5f2e34e6c9
|
[booster] implement Gemini plugin (#3352)
* [booster] add gemini plugin
* [booster] update docstr
* [booster] gemini plugin add coloparam convertor
* [booster] fix coloparam convertor
* [booster] fix gemini plugin device
* [booster] add gemini plugin test
* [booster] gemini plugin ignore sync bn
* [booster] skip some model
* [booster] skip some model
* [booster] modify test world size
* [booster] modify test world size
* [booster] skip test
|
2023-03-31 16:06:13 +08:00 |
|
Frank Lee
|
73d3e4d309
|
[booster] implemented the torch ddd + resnet example (#3232)
* [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 (#3180)
* [booster] added the plugin base and torch ddp plugin
* polish code
* polish code
* polish code
|
2023-03-21 17:39:30 +08:00 |