Pytorch 官方预训练模型的安装和使用

官方预训练模型与torchvision包相关

导入预训练模型语句

import trochvision.models as models
resnet18 = models.resnet18(pretrained=True)

若本地尚未下载该模型信息，则会自动下载，但是该方法会因网速问题下载失败；

因此，选择采用离线安装后直接使用的方法。

首先，找到torchvision.models的安装位置（C:\Users\DELL\miniconda3\envs\PYT\Lib\site-packages\torchvision\models）

以resnet为例，打开resnet.py文件：

以resnet18为例，点击“https://download.pytorch.org/models/resnet18-5c106cde.pth”会直接下载“resnet18-5c106cde.pth”文件

将其放置“C:\Users\DELL\miniconda3\envs\PYT\Lib\site-packages\torchvision\models”文件中，即可直接使用