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

发布时间:2021-05-26 11:05:31 作者:Leah
来源:亿速云 阅读:600

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

推荐阅读:
  1. keras如何实现调用训练模型和去掉全连接层
  2. Pytorch中VGG如何实现修改最后一层FC

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

pytorch resnet

上一篇:使用layui怎么修改table单元格事件的值

下一篇:如何修改nginx站点的根目录

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》