怎么在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模型全连接层就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。

另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。

名称栏目：怎么在Pytorch中修改ResNet模型全连接层-创新互联
URL链接：http://mswzjz.cn/article/ceigis.html