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

Pytorch 官方预训练模型的安装和使用_第1张图片

以resnet为例,打开resnet.py文件:

Pytorch 官方预训练模型的安装和使用_第2张图片

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

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

你可能感兴趣的:(Pytorch)