mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-21 14:13:45 +00:00
5f2e34e6c9
* [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
6 lines
173 B
Python
6 lines
173 B
Python
from .gemini_plugin import GeminiPlugin
|
|
from .plugin_base import Plugin
|
|
from .torch_ddp_plugin import TorchDDPPlugin
|
|
|
|
__all__ = ['Plugin', 'TorchDDPPlugin', 'GeminiPlugin']
|