mirror of https://github.com/hpcaitech/ColossalAI
10 lines
281 B
Python
10 lines
281 B
Python
|
from .parallel_1d import *
|
||
|
|
from .parallel_2d import *
|
||
|
|
from .parallel_2p5d import *
|
||
|
|
from .parallel_3d import *
|
||
|
|
from .parallel_sequence import *
|
||
|
|
from .parallel_vision_transformer import *
|
||
|
|
from .vanilla_resnet import *
|
||
|
|
from .vanilla_vision_transformer import *
|
||
|
|
from .wrapper import *
|