https://github.com/pytorch/vision/tree/master/torchvision/models
几乎所有的常用预训练模型都在这里面
总结下各种模型的下载地址:
1 Resnet: 2 3 model_urls = { 4 \'resnet18\': \'https://download.pytorch.org/models/resnet18-5c106cde.pth\', 5 \'resnet34\': \'https://download.pytorch.org/models/resnet34-333f7ec4.pth\', 6 \'resnet50\': \'https://download.pytorch.org/models/resnet50-19c8e357.pth\', 7 \'resnet101\': \'https://download.pytorch.org/models/resnet101-5d3b4d8f.pth\', 8 \'resnet152\': \'https://download.pytorch.org/models/resnet152-b121ed2d.pth\', 9 } 10 11 inception: 12 13 model_urls = { 14 # Inception v3 ported from TensorFlow 15 \'inception_v3_google\': \'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth\', 16 } 17 18 Densenet: 19 20 model_urls = { 21 \'densenet121\': \'https://download.pytorch.org/models/densenet121-a639ec97.pth\', 22 \'densenet169\': \'https://download.pytorch.org/models/densenet169-b2777c0a.pth\', 23 \'densenet201\': \'https://download.pytorch.org/models/densenet201-c1103571.pth\', 24 \'densenet161\': \'https://download.pytorch.org/models/densenet161-8d451a50.pth\', 25 } 26 27 28 29 Alexnet: 30 31 model_urls = { 32 \'alexnet\': \'https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth\', 33 } 34 35 vggnet: 36 37 model_urls = { 38 \'vgg11\': \'https://download.pytorch.org/models/vgg11-bbd30ac9.pth\', 39 \'vgg13\': \'https://download.pytorch.org/models/vgg13-c768596a.pth\', 40 \'vgg16\': \'https://download.pytorch.org/models/vgg16-397923af.pth\', 41 \'vgg19\': \'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth\', 42 \'vgg11_bn\': \'https://download.pytorch.org/models/vgg11_bn-6002323d.pth\', 43 \'vgg13_bn\': \'https://download.pytorch.org/models/vgg13_bn-abd245e5.pth\', 44 \'vgg16_bn\': \'https://download.pytorch.org/models/vgg16_bn-6c64b313.pth\', 45 \'vgg19_bn\': \'https://download.pytorch.org/models/vgg19_bn-c79401a0.pth\', 46 }