github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/booster/plugin/__init__.py

6 lines
173 B
Python
Raw Normal View History

[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 08:06:13 +00:00
from .gemini_plugin import GeminiPlugin
[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 09:39:30 +00:00
from .plugin_base import Plugin
from .torch_ddp_plugin import TorchDDPPlugin
[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 08:06:13 +00:00
__all__ = ['Plugin', 'TorchDDPPlugin', 'GeminiPlugin']