from enum import Enum


class ModelName(Enum):
    RESNET18 = "resnet18"
    VGG11 = "vgg11"
    VGG13 = "vgg13"
    VGG16 = "vgg16"
    VGG19 = "vgg19"
    SHALLOWNET = "shallownet"
    LINEAR_MODEL = "linear_model"
