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	…
torch_ddp_plugin.py	[booster] fixed the torch ddp plugin with the new checkpoint api (#3442)	2023-04-06 09:43:51 +08:00