ColossalAI/tests/test_trainer/test_pipeline/model/__init__.py

3 lines
56 B
Python

from .layers import *
from .resnet import VanillaResNet