58、resnet50 模型下载

上一节配置好 python 环境之后，本节开始下载 resnet50 训练好的模型。我们使用以下代码完成 resnet50 的模型下载。

import numpy as np
from torchvision import models
import torch

resnet50 = torch.hub.load('pytorch/vision:v0.10.0', 'resnet50', pretrained=True)
resnet50.eval()
print(resnet50)

代码链接在：practice/model/resnet50_parser.py · iwaihou/cv_learning_from_scratch - Gitee.com

可以先看前 7 行，其余的暂时不需要看。

torch.hub.load

torch.hub.load 是 PyTorch 库中的一个函数，用于从 PyTorch Hub 中加载预训练的模型，示例中通过这个接口，下载了 resnet50 这个模型。

在这个接口中，pretrained = True 说明下载的模型是已经预训练好的，也就是模型中包含有训练好的参数。

resnet50.eval() 表示将模型设置为推理模式。所谓推理模式也就是使用该模型来进行推理而非训练。