pytorch 修改预训练model实例
更新时间：2020年01月18日 15:38:52 作者：whut_ldz
今天小编就为大家分享一篇pytorch 修改预训练model实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
我就废话不多说了，直接上代码吧！
class Net(nn.Module): def __init__(self , model): super(Net, self).__init__() #取掉model的后两层 self.resnet_layer = nn.Sequential(*list(model.children())[:-2]) self.transion_layer = nn.ConvTranspose2d(2048, 2048, kernel_size=14, stride=3) self.pool_layer = nn.MaxPool2d(32) self.Linear_layer = nn.Linear(2048, 8) def forward(self, x): x = self.resnet_layer(x) x = self.transion_layer(x) x = self.pool_layer(x) x = x.view(x.size(0), -1) x = self.Linear_layer(x) return x
resnet = models.resnet50(pretrained=True) model = Net(resnet)
以上这篇pytorch 修改预训练model实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
Python使用自带的base64库进行base64编码和解码
在Python中,处理数据的编码和解码是数据传输和存储中非常普遍的需求,其中,Base64是一种常用的编码方案,本文我将详细介绍如何使用Python的base64库进行Base64编码和解码,需要的朋友可以参考下2025-04-04
这篇文章主要介绍了用Cython加速Python到“起飞”，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
这篇文章主要介绍了Python基于execjs运行js过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
这篇文章主要介绍了python 写函数在一定条件下需要调用自身时的写法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
在本篇文章里小编给大家分享了关于python把转列表为集合的方法以及相关实例内容，有兴趣的朋友们学习下。2019-06-06
Python基于Tkinter的HelloWorld入门实例
这篇文章主要介绍了Python基于Tkinter的HelloWorld入门实例,以一个简单实例分析了Python中Tkinter模块的使用技巧,需要的朋友可以参考下2015-06-06
这篇文章主要介绍了python Socket网络编程实现C/S模式和P2P，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-06-06
这篇文章主要为大家详细介绍了python实现抠图给证件照换背景源码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-08-08
这篇文章主要介绍了python Django的web开发实例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
这篇文章主要为大家详细介绍了Python 查询学生的平均分数的实现,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下2019-04-04
最新评论