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官方提供的模型库，用户都可以在加载预训练模型后进一步微调或使用该模型进行推理。