mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-11 14:43:10 +00:00
e7f3bed2d3
* [booster] added the plugin base and torch ddp plugin * polish code * polish code * polish code
5 lines
117 B
Python
5 lines
117 B
Python
from .plugin_base import Plugin
|
|
from .torch_ddp_plugin import TorchDDPPlugin
|
|
|
|
__all__ = ['Plugin', 'TorchDDPPlugin']
|