import torchvision.models as models

imagenet_models = [
    models.resnet18, models.resnet34, models.resnet50, models.resnet101,
    models.wide_resnet50_2, models.wide_resnet101_2,
    models.vgg11, models.vgg11_bn, models.vgg16, models.vgg16_bn,
    models.mobilenet_v2, models.shufflenet_v2_x1_0,
    models.densenet121, models.densenet169, models.densenet201,
    models.vit_b_32, models.vit_b_16, models.vit_l_32, models.vit_l_16,
    models.swin_t, models.swin_s, models.swin_v2_t, models.swin_v2_s, 
]

cifar10_models = [
    'cifar10_resnet20', 'cifar10_resnet32', 'cifar10_resnet44', 'cifar10_resnet56',
    'cifar10_vgg11_bn', 'cifar10_vgg13_bn', 'cifar10_vgg16_bn', 'cifar10_vgg19_bn',
    'cifar10_mobilenetv2_x0_5', 'cifar10_mobilenetv2_x0_75', 'cifar10_mobilenetv2_x1_0', 'cifar10_mobilenetv2_x1_4',
    'cifar10_shufflenetv2_x0_5', 'cifar10_shufflenetv2_x1_0', 'cifar10_shufflenetv2_x1_5', 'cifar10_shufflenetv2_x2_0',
    'cifar10_repvgg_a0', 'cifar10_repvgg_a1', 'cifar10_repvgg_a2',
    'binyxuCUHK/cifar10-vit-b-32',
    'binyxuCUHK/cifar10-vit-b-16',
    'binyxuCUHK/cifar10-vit-l-32',
    'binyxuCUHK/cifar10-vit-l-16',
    'binyxuCUHK/cifar10-swin-b',
    'binyxuCUHK/cifar10-swin-l',
    'binyxuCUHK/cifar10-swin-v2-b',
    'binyxuCUHK/cifar10-swin-v2-l',
]

cifar100_models = [
    'cifar100_resnet20', 'cifar100_resnet32', 'cifar100_resnet44', 'cifar100_resnet56',
    'cifar100_vgg11_bn', 'cifar100_vgg13_bn', 'cifar100_vgg16_bn', 'cifar100_vgg19_bn',
    'cifar100_mobilenetv2_x0_5', 'cifar100_mobilenetv2_x0_75', 'cifar100_mobilenetv2_x1_0', 'cifar100_mobilenetv2_x1_4',
    'cifar100_shufflenetv2_x0_5', 'cifar100_shufflenetv2_x1_0', 'cifar100_shufflenetv2_x1_5', 'cifar100_shufflenetv2_x2_0',
    'cifar100_repvgg_a0', 'cifar100_repvgg_a1', 'cifar100_repvgg_a2',
    'binyxuCUHK/cifar100-vit-b-32',
    'binyxuCUHK/cifar100-vit-b-16',
    'binyxuCUHK/cifar100-vit-l-32',
    'binyxuCUHK/cifar100-vit-l-16',
    'binyxuCUHK/cifar100-swin-b',
    'binyxuCUHK/cifar100-swin-l',
    'binyxuCUHK/cifar100-swin-v2-b',
    'binyxuCUHK/cifar100-swin-v2-l',
]