# 加载 ResNet50 模型 model = models.resnet50(pretrained=True) # 修改全连接层以适应50类别分类任务 model.fc = nn.Linear(model.fc.in_features, 50)
