PyTorch Torchvision Models

Model Acc@1 Acc@5
AlexNet 56.522 79.066
VGG-11 69.020 88.628
VGG-13 69.928 89.246
VGG-16 71.592 90.382
VGG-19 72.376 90.876
VGG-11 with batch normalization 70.370 89.810
VGG-13 with batch normalization 71.586 90.374
VGG-16 with batch normalization 73.360 91.516
VGG-19 with batch normalization 74.218 91.842
ResNet-18 69.758 89.078
ResNet-34 73.314 91.420
ResNet-50 76.130 92.862
ResNet-101 77.374 93.546
ResNet-152 78.312 94.046
SqueezeNet 1.0 58.092 80.420
SqueezeNet 1.1 58.178 80.624
Densenet-121 74.434 91.972
Densenet-169 75.600 92.806
Densenet-201 76.896 93.370
Densenet-161 77.138 93.560
Inception v3 77.294 93.450
GoogleNet 69.778 89.530
ShuffleNet V2 x1.0 69.362 88.316
ShuffleNet V2 x0.5 60.552 81.746
MobileNet V2 71.878 90.286
MobileNet V3 Large 74.042 91.340
MobileNet V3 Small 67.668 87.402
ResNeXt-50-32x4d 77.618 93.698
ResNeXt-101-32x8d 79.312 94.526
Wide ResNet-50-2 78.468 94.086
Wide ResNet-101-2 78.848 94.284
MNASNet 1.0 73.456 91.510
MNASNet 0.5 67.734 87.490
EfficientNet-B0 77.692 93.532
EfficientNet-B1 78.642 94.186
EfficientNet-B2 80.608 95.310
EfficientNet-B3 82.008 96.054
EfficientNet-B4 83.384 96.594
EfficientNet-B5 83.444 96.628
EfficientNet-B6 84.008 96.916
EfficientNet-B7 84.122 96.908
regnet_x_400mf 72.834 90.950
regnet_x_800mf 75.212 92.348
regnet_x_1_6gf 77.040 93.440
regnet_x_3_2gf 78.364 93.992
regnet_x_8gf 79.344 94.686
regnet_x_16gf 80.058 94.944
regnet_x_32gf 80.622 95.248
regnet_y_400mf 74.046 91.716
regnet_y_800mf 76.420 93.136
regnet_y_1_6gf 77.950 93.966
regnet_y_3_2gf 78.948 94.576
regnet_y_8gf 80.032 95.048
regnet_y_16gf 80.424 95.240
regnet_y_32gf 80.878 95.340

Alexnet

https://arxiv.org/abs/1404.5997

  • torchvision.models.alexnet

VGG

https://arxiv.org/abs/1409.1556

  • torchvision.models.vgg11

  • torchvision.models.vgg11_bn

  • torchvision.models.vgg13

  • torchvision.models.vgg13_bn

  • torchvision.models.vgg16

  • torchvision.models.vgg16_bn

  • torchvision.models.vgg19

  • torchvision.models.vgg19_bn


ResNet

https://arxiv.org/abs/1512.03385

  • torchvision.models.resnet18
  • torchvision.models.resnet34
  • torchvision.models.resnet50
  • torchvision.models.resnet101
  • torchvision.models.resnet152

SqueezeNet

https://arxiv.org/abs/1602.07360

  • torchvision.models.squeezenet1_0
  • torchvision.models.squeezenet1_1

DenseNet

https://arxiv.org/abs/1608.06993

  • torchvision.models.densenet121
  • torchvision.models.densenet169
  • torchvision.models.densenet161
  • torchvision.models.densenet201

Inception v3

https://arxiv.org/abs/1512.00567

  • torchvision.models.inception_v3
    requires scipy to be installed

GoogLeNet

https://arxiv.org/abs/1409.4842

  • torchvision.models.googlenet
    requires scipy to be installed

ShuffleNet v2

https://arxiv.org/abs/1807.11164

  • torchvision.models.shufflenet_v2_x0_5
  • torchvision.models.shufflenet_v2_x1_0
  • torchvision.models.shufflenet_v2_x1_5
  • torchvision.models.shufflenet_v2_x2_0

MobileNet v2

https://arxiv.org/abs/1801.04381

  • torchvision.models.mobilenet_v2

MobileNet v3

https://arxiv.org/abs/1905.02244

  • torchvision.models.mobilenet_v3_large
  • torchvision.models.mobilenet_v3_small

ResNext

https://arxiv.org/abs/1611.05431

  • torchvision.models.resnext50_32x4d
  • torchvision.models.resnext101_32x8d

Wide ResNet

  • torchvision.models.wide_resnet50_2
  • torchvision.models.wide_resnet101_2

MNASNet

https://arxiv.org/abs/1807.11626

  • torchvision.models.mnasnet0_5
  • torchvision.models.mnasnet0_75
  • torchvision.models.mnasnet1_0
  • torchvision.models.mnasnet1_3

EfficientNet

https://arxiv.org/abs/1905.11946

  • torchvision.models.efficientnet_b0
  • torchvision.models.efficientnet_b1
  • torchvision.models.efficientnet_b2
  • torchvision.models.efficientnet_b3
  • torchvision.models.efficientnet_b4
  • torchvision.models.efficientnet_b5
  • torchvision.models.efficientnet_b6
  • torchvision.models.efficientnet_b7

RegNet

https://arxiv.org/abs/2003.13678

  • torchvision.models.regnet_y_400mf
  • torchvision.models.regnet_y_800mf
  • torchvision.models.regnet_y_1_6gf
  • torchvision.models.regnet_y_3_2gf
  • torchvision.models.regnet_y_8gf
  • torchvision.models.regnet_y_16gf
  • torchvision.models.regnet_y_32gf
  • torchvision.models.regnet_x_400mf
  • torchvision.models.regnet_x_800mf
  • torchvision.models.regnet_x_1_6gf
  • torchvision.models.regnet_x_3_2gf
  • torchvision.models.regnet_x_8gf
  • torchvision.models.regnet_x_16gf