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
..
mixed_precision	[booster] implemented the torch ddd + resnet example (#3232)	2023-03-27 10:24:14 +08:00
plugin	[booster] implement Gemini plugin (#3352)	2023-03-31 16:06:13 +08:00
__init__.py	[booster] implemented the torch ddd + resnet example (#3232)	2023-03-27 10:24:14 +08:00
accelerator.py	[booster] added the accelerator implementation (#3159)	2023-03-20 13:59:24 +08:00
booster.py	[booster] implemented the torch ddd + resnet example (#3232)	2023-03-27 10:24:14 +08:00