本文主要是介绍本地下载预训练模型(涉及内容:Resnet等预训练模型地址,以resnet34为例下载预训练模型并移动到指定地址),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
运行如下命令:
finetune_net.features = torchvision.models.resnet34(pretrained=True)
一直报错。
离线下载模型的预训练模型地址
复制需要下载的模型地址,粘贴到浏览器地址栏中下载,各种模型的下载地址如下:
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',}2. inception:model_urls = {Inception v3 ported from TensorFlow'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth',}3. 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',}4. Alexnet:model_urls = {'alexnet': 'https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth',
}5. 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',}
下载预训练模型(以resnet34为例)
浏览器输入:
https://download.pytorch.org/models/resnet34-333f7ec4.pth
移动预训练模型到 ~/.cache/torch/hub/checkpoints/
我的linux自动下载到了Downloads文件夹下。
下载好的文件名为:resnet34-333f7ec4.pth
程序默认读取~/.cache/torch/hub/checkpoints/文件下的 预训练模型
输入如下命令,将该预训练模型移动到~/.cache/torch/hub/checkpoints/
mv ~/Downloads/resnet34-333f7ec4.pth ~/.cache/torch/hub/checkpoints/
参考链接
Pytorch离线下载并使用torchvision.models预训练模型
https://blog.csdn.net/qq_37555071/article/details/107789783
这篇关于本地下载预训练模型(涉及内容:Resnet等预训练模型地址,以resnet34为例下载预训练模型并移动到指定地址)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!