在Torch中使用預訓練模型有兩種常見的方法:
import torch
model = torch.hub.load('pytorch/vision:v0.9.0', 'resnet18', pretrained=True)
import torch
import torchvision.models as models
model = models.vgg16(pretrained=True)
無論是使用Torch Hub還是PyTorch官方提供的模型庫,用戶都可以在加載預訓練模型后進一步微調或使用該模型進行推理。