在PyTorch中使用预训练的模型进行迁移学习可以通过以下步骤实现：
import torchvision.models as models
# Load pre-trained ResNet-50 model
model = models.resnet50(pretrained=True)
import torch.nn as nn
# Modify the last layer of the model
num_ftrs = model.fc.in_features
model.fc = nn.Linear(num_ftrs, num_classes) # num_classes为新任务的类别数
import torch.optim as optim
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9)
# 训练代码
通过以上步骤，您可以在PyTorch中使用预训练的模型进行迁移学习。
