Untitled

Anonymous
plain_text
01/04/2025 3:48 AM
224 B
21
Indexable
# 加载 ResNet50 模型
model = models.resnet50(pretrained=True)
# 修改全连接层以适应50类别分类任务
model.fc = nn.Linear(model.fc.in_features, 50)
Editor is loading...
Leave a Comment