解决pytorch预训练模型下载速度慢的问题 |
您所在的位置：网站首页 › 英伟达下载速度太慢 › 解决pytorch预训练模型下载速度慢的问题 |
解决pytorch预训练模型下载速度慢的问题
|
在远程跑代码的时候，我需要 from torchvision.models.resnet import resnet50然而远程服务器上下载太慢太慢了，有时还会下载进度突然不动了，这……心累啊…… 解决方法： 1. 去以下网址下载你需要的模型： Resnet: model_urls = { 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', 'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth', 'resnet101': 'https://download.pytorch.org/models/resnet101-5d3b4d8f.pth', 'resnet152': 'https://download.pytorch.org/models/resnet152-b121ed2d.pth', } inception: model_urls = { # Inception v3 ported from TensorFlow 'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth', } Densenet: model_urls = { 'densenet121': 'https://download.pytorch.org/models/densenet121-a639ec97.pth', 'densenet169': 'https://download.pytorch.org/models/densenet169-b2777c0a.pth', 'densenet201': 'https://download.pytorch.org/models/densenet201-c1103571.pth', 'densenet161': 'https://download.pytorch.org/models/densenet161-8d451a50.pth', } Alexnet: model_urls = { 'alexnet': 'https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth', } vggnet: model_urls = { 'vgg11': 'https://download.pytorch.org/models/vgg11-bbd30ac9.pth', 'vgg13': 'https://download.pytorch.org/models/vgg13-c768596a.pth', 'vgg16': 'https://download.pytorch.org/models/vgg16-397923af.pth', 'vgg19': 'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth', 'vgg11_bn': 'https://download.pytorch.org/models/vgg11_bn-6002323d.pth', 'vgg13_bn': 'https://download.pytorch.org/models/vgg13_bn-abd245e5.pth', 'vgg16_bn': 'https://download.pytorch.org/models/vgg16_bn-6c64b313.pth', 'vgg19_bn': 'https://download.pytorch.org/models/vgg19_bn-c79401a0.pth', }2. torch在加载模型时候首先检查本地缓存是否已经存在模型，直接将下载的文件放到 .cache/torch/checkpoints/ 目录中。 我的是 /home/zy/.cache/torch/checkpoints/
参考资料：pytorch预训练模型的下载地址以及解决下载速度慢的方法 |
【本文地址】
今日新闻 |
推荐新闻 |