mirror of https://github.com/hpcaitech/ColossalAI
7 lines
210 B
Python
7 lines
210 B
Python
|
from .base import Strategy
|
||
|
|
from .colossalai import ColossalAIStrategy
|
||
|
|
from .ddp import DDPStrategy
|
||
|
|
from .naive import NaiveStrategy
|
||
|
|
|
||
|
|
__all__ = ['Strategy', 'NaiveStrategy', 'DDPStrategy', 'ColossalAIStrategy']
|