SDSC6001_HW3 / models /__init__.py
MingLi
add ViT and InceptionV3
16456f4
from .VGG16Classifier import VGG16Classifier
from .ResNetClassifier import ResNet50Classifier, ResNet101Classifier
from .InceptionV3Classifier import InceptionV3Classifier
from .VITClassifier import ViTLargeClassifier
__all__ = [
"VGG16Classifier",
"ResNet50Classifier",
"ResNet101Classifier",
"InceptionV3Classifier",
"ViTLargeClassifier",
]