Files
ColossalAI/model_zoo/vit/__init__.py
2021-10-28 18:21:23 +02:00

3 lines
54 B
Python

from .parallel_2d import *
from .parallel_3d import *