位置: IT常识 - 正文

pytorch对网络层的增,删, 改, 修改预训练模型结构(pytorch自定义网络层)

编辑:rootadmin
pytorch对网络层的增,删, 改, 修改预训练模型结构 #下载模型参数model.load_state_dict(torch.load('model.pth'))#再加载网络的参数torch.load('model.pth')是获得网络参数

推荐整理分享pytorch对网络层的增,删, 改, 修改预训练模型结构(pytorch自定义网络层),希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:pytorch怎么训练网络,pytorch 网络,pytorch网络搭建,pytorch定义网络,pytorch cnn网络,pytorch bp网络,pytorch输出网络结构,pytorch cnn网络,内容如对您有帮助,希望把文章链接给更多的朋友!

1.我们使用vgg11网络做示例, 看一下网络结构:

加载本地的模型:

vgg16 = models.vgg16(pretrained=False)#打印出预训练模型的参数vgg16.load_state_dict(torch.load('vgg16-397923af.pth'))

加载库中的模型

import torchimport torch.nn as nnfrom torchvision import modelsnet = models.vgg11(pretrained=True)print(net)

1)(1). 在网络中添加一层:

net网络是一个树型结构, net下面有三个结点,分别是(features, avgpoll, classifier), 我们先在features结点添加一层’lastlayer'层

net.features.add_module('lastlayer', nn.Conv2d(512,512, kernel_size=3, stride=1, padding=1))print(net) 1)(2). 在classifier结点添加一个线性层:net.classifier.add_module('Linear', nn.Linear(1000, 10))print(net)

2)(1)修改网络中的某一层(features 结点举例):net.features[8] = nn.Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))print(net)

 2)(2)修改网络中的某一层(classifier结点举例):net.classifier[6] = nn.Linear(1000, 5)print(net)pytorch对网络层的增,删, 改, 修改预训练模型结构(pytorch自定义网络层)

注意: 这里我尝试对Linear这一层进行更新, 但是Linear名字是字符串, 提取不出来,所以应该在之前添加网络时候, 名字不要取字符串, 否则会报错  ‘  'str' object cannot be interpreted as an integer’。 

 3)(1) 网络层的删除(features举例) classifier结点的操作相同。

直接使用nn.Sequential()对改层设置为空即可

net.features[13] = nn.Sequential()print(net)

 4)冻结网络中某些层 (直接使该层的requires_grad = False)即可, 这样在反向传播的时候,不会更新该层的参数#冻结指定层的预训练参数:net.feature[26].weight.requires_grad = False5). 第二种对网络结构的操作方法:net.features = nn.Sequential(*list(net.features.children())[:-4])

可以看到后面4层被去除了, 就是说可以使用列表切片的方法来删除网络层

net.classifier 对应 net.classifier.children()

net.features 对应 net.features.children()

  1. 先加载网络结构

自己的模型, model的类要有定义才可以, 如果在其他.py文件中,可以导入文件,然后用文件中的类实例化对象。model = torch.load(PATH)

 2.再加载网络参数

#下载模型参数

model.load_state_dict(torch.load('model.pth'))#再加载网络的参数torch.load('model.pth')是获得网络参数
本文链接地址:https://www.jiuchutong.com/zhishi/299377.html 转载请保留说明!

上一篇:vue 项目兼容 IE 浏览器(vue项目兼容ie9以上浏览器)

下一篇:【CSS】CSS 背景设置 ⑨ ( 背景半透明设置 )(css背景图)

  • 从QQ广告看互联网广告新趋势(qq上的广告可信吗)

    从QQ广告看互联网广告新趋势(qq上的广告可信吗)

  • ps怎么把字体变成空心字体(ps怎么把字体变清晰)

    ps怎么把字体变成空心字体(ps怎么把字体变清晰)

  • 计算器怎么调声音(计算器怎么调声音和音量)

    计算器怎么调声音(计算器怎么调声音和音量)

  • 央视影音版hd版区别(央视影音hd正式版本)

    央视影音版hd版区别(央视影音hd正式版本)

  • 拼多多怎么提升访客量(拼多多怎么提升权重)

    拼多多怎么提升访客量(拼多多怎么提升权重)

  • airpods pro充电敲击没反应(airpodspro充电敲两次)

    airpods pro充电敲击没反应(airpodspro充电敲两次)

  • 冰箱最上面一层叫什么(冰箱最上面一层不制冷怎么回事)

    冰箱最上面一层叫什么(冰箱最上面一层不制冷怎么回事)

  • 苹果8p卡贴机能要吗(苹果8p卡贴机能升级系统吗)

    苹果8p卡贴机能要吗(苹果8p卡贴机能升级系统吗)

  • 对siri说关闭网络会怎样(对siri说关闭网络)

    对siri说关闭网络会怎样(对siri说关闭网络)

  • qq标识小海螺是什么(qq海螺是什么意思)

    qq标识小海螺是什么(qq海螺是什么意思)

  • 双耳蓝牙耳机可以分开使用吗(双耳蓝牙耳机可以分开连接两个手机吗)

    双耳蓝牙耳机可以分开使用吗(双耳蓝牙耳机可以分开连接两个手机吗)

  • a1370苹果笔记本电脑是哪一年的(A1370苹果笔记本win7 摄像头驱动)

    a1370苹果笔记本电脑是哪一年的(A1370苹果笔记本win7 摄像头驱动)

  • qq删了再下载聊天记录还在吗(qq删了再下载聊天记录没有了怎么办)

    qq删了再下载聊天记录还在吗(qq删了再下载聊天记录没有了怎么办)

  • c11和dtp的区别(7pc11和dtp的区别)

    c11和dtp的区别(7pc11和dtp的区别)

  • 爱奇艺黄金会员可以几个人同时登录吗(爱奇艺黄金会员是什么意思)

    爱奇艺黄金会员可以几个人同时登录吗(爱奇艺黄金会员是什么意思)

  • ps怎么调整图片方向(ps怎么调整图片角度)

    ps怎么调整图片方向(ps怎么调整图片角度)

  • 卡贴iphone怎么开热点(卡贴怎么打开)

    卡贴iphone怎么开热点(卡贴怎么打开)

  • 手机怎么拍流星(手机怎么拍流星雨的照片)

    手机怎么拍流星(手机怎么拍流星雨的照片)

  • 手机瀑布屏是什么意思(瀑布屏是啥)

    手机瀑布屏是什么意思(瀑布屏是啥)

  • 苹果11怎么设置电量显示(苹果11怎么设置密码锁屏)

    苹果11怎么设置电量显示(苹果11怎么设置密码锁屏)

  • 小米note3什么时候上市的(小米note3再战几年)

    小米note3什么时候上市的(小米note3再战几年)

  • 360压缩怎么压缩文件(360压缩怎么压缩到指定大小)

    360压缩怎么压缩文件(360压缩怎么压缩到指定大小)

  • 水印相机怎么没录像功能吗(水印相机怎么没时间)

    水印相机怎么没录像功能吗(水印相机怎么没时间)

  • 从网上下载软件时使用的网络服务类型是(从网上下载软件时使用的网络类型)

    从网上下载软件时使用的网络服务类型是(从网上下载软件时使用的网络类型)

  • 成功app开发有哪些误区(app开发好后需要做什么)

    成功app开发有哪些误区(app开发好后需要做什么)

  • 苹果查找我的iphone在哪(苹果查找我的iphone入口)

    苹果查找我的iphone在哪(苹果查找我的iphone入口)

  • deepin20终端远程管理服务器的方法(deepin 终端命令)

    deepin20终端远程管理服务器的方法(deepin 终端命令)

  • macOS Big Sur 11.3 开发者预览版/公测版 Beta 6正式发布

    macOS Big Sur 11.3 开发者预览版/公测版 Beta 6正式发布

  • 偶然所得个人所得税会计分录
  • 发票系统口令锁定
  • 小规模纳税人怎么变成一般纳税人
  • 小规模附加税减半征收吗
  • 发票认证没有信用怎么办
  • 技术使用费所得税怎么算
  • 融资租赁可以确认为资产吗
  • 费用已付发票未到的预算会计分录
  • 事业单位下属企业员工转事业编
  • 工伤私了赔偿
  • 保障房异地建设费缴纳契税吗?
  • 计提租金分录怎么处理
  • 关于转让费的问题
  • 企业为员工承担个人所得税
  • 日用品专用发票可以抵税嘛
  • 个贷系统平账专户利息分录
  • 金税盘不交费会停用吗
  • 以前年度未申报的专项附加扣除
  • 股东和原始股的区别
  • 股东撤资退出
  • windows10专业
  • 终于找到修改mac的方法了!
  • 速动比率计算公式多少合适
  • 电子承兑汇票贴现怎么做账
  • 成品油属于什么费用
  • 民办非企业单位是私立还是公立
  • mysql高级语句
  • 21年最新cpu
  • 转账凭证摘要怎么写原材料的
  • php解密在线
  • 搜索框无法搜索内容
  • 采购涉及哪些部门
  • 供热企业供热费收入的核算
  • vue3 global
  • 凭证字号怎么写
  • html下划线怎么设置
  • 微信小程序在哪里找?
  • 深度学习之快速实现数据集增强的方法
  • js-cookie vue
  • 采矿权价款怎么做账
  • 不同情况不同处理的句子
  • 政府补助可以挂账吗
  • 企业应收账款的主要内容
  • 进出口总额用什么字母表示
  • python numpy 删除元素
  • php接口技术
  • sql server递归查询
  • 各部门和单位按照国家有关规定收取或取得
  • 广东高速公路过路费收费标准
  • 开工程款发票有什么规定?
  • 广告制作费怎么入账
  • 预付账款是什么意思大白话
  • 汇算清缴要退税1750元,不想退要调平怎么调?
  • 纳税人在工作
  • 加盟费开发票
  • 未开票收入怎么报税
  • 软件折旧从什么时候算
  • 汽车几年折旧年限
  • sql server数据
  • mysql根据另一张表更新
  • xp系统找不到指定的路径怎么办
  • Linux配置防火墙端口
  • win10预览版好吗
  • win8系统怎么关掉开机密码
  • win8系统设置锁屏密码
  • win7在开始菜单右击
  • windows磁盘工具
  • win10系统附件游戏被删除
  • windows7打开网络连接
  • win8系统安装的软件在哪里
  • 新手学做ppt
  • macos unity
  • 模型图怎么画
  • jqueryui
  • 安卓开发界面ui设计器
  • android模拟器无法运行
  • android toolbar menu
  • 税务徽标含义
  • 城镇土地使用税暂行条例
  • 怎样通过发票号码查询电子发票
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设