怎么在Pytorch中修改ResNet模型全连接层

这篇文章给大家介绍怎么在Pytorch中修改ResNet模型全连接层，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。

最后一层的输出改一下，不需要加载前面层的权重，方法如下：

model = torchvision.models.resnet18(pretrained=False)
num_fc_ftr = model.fc.in_features
model.fc = torch.nn.Linear(num_fc_ftr, 224)
model = nn.DataParallel(model, device_ids=config.gpus).to(device)

首先模型结构是必须要传入的，然后把最后一层的输出改为自己所需的数目

关于怎么在Pytorch中修改ResNet模型全连接层就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。