Torch中的预训练模型怎么使用

在Torch中使用预训练模型有两种常见的方法：

1. 使用Torch Hub：Torch Hub是一个官方的模型库，包含了一些常见的预训练模型，用户可以通过Torch Hub加载这些模型。例如，可以通过以下代码加载一个预训练的ResNet模型：

import torch
model = torch.hub.load('pytorch/vision:v0.9.0', 'resnet18', pretrained=True)

2. 使用PyTorch官方提供的模型库：PyTorch官方提供了一些常见的预训练模型，用户可以通过导入相应的模块来使用这些模型。例如，可以通过以下代码加载一个预训练的VGG模型：

import torch
import torchvision.models as models
model = models.vgg16(pretrained=True)

无论是使用Torch Hub还是PyTorch官方提供的模型库，用户都可以在加载预训练模型后进一步微调或使用该模型进行推理。

亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>