调用torchvision实现resnet的迁移学习

可以使用以下代码进行resnet的迁移学习：

import torchvision.models as models

resnet = models.resnet18(pretrained=True)
# 将最后一个全连接层替换为 num_classes 输出
num_classes = 10
resnet.fc = nn.Linear(resnet.fc.in_features, num_classes)

其中，num_classes 表示分类的类别数，可以根据具体问题进行修改。