Frank Lee 7d8d825681 [booster] fixed the torch ddp plugin with the new checkpoint api (#3442)

2023-04-06 09:43:51 +08:00

..

__init__.py

[booster] implement Gemini plugin (#3352)

2023-03-31 16:06:13 +08:00

gemini_plugin.py

[booster] fixed the torch ddp plugin with the new checkpoint api (#3442)

2023-04-06 09:43:51 +08:00

plugin_base.py

[booster] implemented the torch ddd + resnet example (#3232)

2023-03-27 10:24:14 +08:00

torch_ddp_plugin.py

[booster] fixed the torch ddp plugin with the new checkpoint api (#3442)

2023-04-06 09:43:51 +08:00