码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch预训练模型的下载地址以及解决下载速度慢的方法

时间:2019-03-27 10:36:29      阅读:2857      评论:0      收藏:0      [点我收藏+]

标签:color   预训练模型   class   models   mod   速度   网络   load   ast   

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 }

 

解决下载速度慢的方法:

换移动网络,校园网对于pytorch网站有很大的限速。

pytorch预训练模型的下载地址以及解决下载速度慢的方法

标签:color   预训练模型   class   models   mod   速度   网络   load   ast   

原文地址:https://www.cnblogs.com/ywheunji/p/10605614.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!