调用pytorch库中的torchvision实现resnet的迁移学习

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

import torch
from torchvision import models

resnet50 = models.resnet50(pretrained=True)
for param in resnet50.parameters():
    param.requires_grad = False
    
num_ftrs = resnet50.fc.in_features
resnet50.fc = torch.nn.Linear(num_ftrs, 2) # 根据具体任务修改输出层

# 然后就可以将处理好的数据传入模型进行训练了