要加载预训练模型,首先需要下载预训练模型的参数文件,然后在 PyTorch 中加载这些参数。下面是一个示例代码:
import torch
# 加载预训练模型
model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)
# 设置模型为评估模式
model.eval()
在这个代码中,我们使用 torch.hub.load
函数来加载预训练的 ResNet-18 模型。请注意,我们将 pretrained
参数设置为 True
,以确保加载预训练的参数。最后,我们将模型设置为评估模式