pytorch 修改预训练model实例
我就废话不多说了，直接上代码吧！
classNet(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) defforward(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) returnx
resnet=models.resnet50(pretrained=True) model=Net(resnet)
以上这篇pytorch修改预训练model实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。