File size: 2,082 Bytes
998bb30 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
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',
]
|