位置: 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背景图)

  • 华为pro20什么时候上市的(华为pro20pro)(华为pro20什么时候出的)

    华为pro20什么时候上市的(华为pro20pro)(华为pro20什么时候出的)

  • 华为手环7有来电提醒吗(华为手环7来电不震动解决办法)

    华为手环7有来电提醒吗(华为手环7来电不震动解决办法)

  • 手机怎么看b站注册日期(手机怎么看b站港澳台番剧)

    手机怎么看b站注册日期(手机怎么看b站港澳台番剧)

  • qq音乐绿钻可以听vip歌曲吗(qq音乐绿钻可以送人吗)

    qq音乐绿钻可以听vip歌曲吗(qq音乐绿钻可以送人吗)

  • 华为荣耀8x是5g手机吗(华为荣耀8X是什么型号)

    华为荣耀8x是5g手机吗(华为荣耀8X是什么型号)

  • layout软件是干什么的

    layout软件是干什么的

  • 9代酷睿为什么取消超线程(9代酷睿为什么还那么贵)

    9代酷睿为什么取消超线程(9代酷睿为什么还那么贵)

  • 群被投诉了会封号吗(群被投诉了会封群吗)

    群被投诉了会封号吗(群被投诉了会封群吗)

  • 淘宝客定向计划是什么意思(淘宝客定向计划名称怎么写)

    淘宝客定向计划是什么意思(淘宝客定向计划名称怎么写)

  • qq上一起听歌怎么弄(QQ上一起听歌怎么换歌)

    qq上一起听歌怎么弄(QQ上一起听歌怎么换歌)

  • 苹果7的耳机插孔在哪里(苹果7的耳机插头有一点歪正常吗)

    苹果7的耳机插孔在哪里(苹果7的耳机插头有一点歪正常吗)

  • 华为p40有呼吸灯吗(华为p40呼吸灯在哪里设置)

    华为p40有呼吸灯吗(华为p40呼吸灯在哪里设置)

  • 电话停机多久会被注销(电话停机多久会空号)

    电话停机多久会被注销(电话停机多久会空号)

  • 苹果日版和美版区别(苹果日版和美版质检一样吗)

    苹果日版和美版区别(苹果日版和美版质检一样吗)

  • 手机deliverserver是什么软件(手机号码测吉凶)

    手机deliverserver是什么软件(手机号码测吉凶)

  • qq升一级需要多少天(qq升一级加多少能量)

    qq升一级需要多少天(qq升一级加多少能量)

  • 手机压缩包是什么意思(手机压缩包是什么可以删吗)

    手机压缩包是什么意思(手机压缩包是什么可以删吗)

  • 华为nova5pro与nova6对比(华为nova5pro与nova7se对比)

    华为nova5pro与nova6对比(华为nova5pro与nova7se对比)

  • 相机专业模式s是什么意思(相机专业模式是什么档)

    相机专业模式s是什么意思(相机专业模式是什么档)

  • 亲情付退款退哪了(亲情付退款后不能退回额度吗)

    亲情付退款退哪了(亲情付退款后不能退回额度吗)

  • 小米cc9e支持nfc吗(小米cc9e支持5g网络吗)

    小米cc9e支持nfc吗(小米cc9e支持5g网络吗)

  • 拼多多怎么扫码找商品(拼多多怎么扫码取件)

    拼多多怎么扫码找商品(拼多多怎么扫码取件)

  • 小米8屏幕颗粒感严重(小米8屏幕抗刮怎么样)

    小米8屏幕颗粒感严重(小米8屏幕抗刮怎么样)

  • Vue.js安装与创建默认项目(详细步骤)(vue.js如何安装)

    Vue.js安装与创建默认项目(详细步骤)(vue.js如何安装)

  • 小规模纳税人转一般纳税人
  • 工资里个人代扣款是什么
  • 收客户刷卡手续费时客户不愿意缴费,如何处理
  • 经营性现金净流量为负数怎么解释
  • 减免税款的会计分录摘要
  • 个人所得税年度汇算时间
  • 成本票和进项票怎么区分
  • 税额四舍五入的计算公式
  • 工会筹备金需不需要计提
  • 什么叫备抵科目,什么叫过渡科目
  • 资产盘点的步骤是什么,需要注意哪些细节
  • 结存材料应分摊什么
  • 建筑业异地预缴税款怎么抵扣
  • 本月做未开票收入,下月开票怎么做账
  • 委托其他公司开票收款
  • 土地评估费计入什么会计科目
  • 展会门票费计入什么科目
  • 原材料验收的重要性
  • 福利企业退税的资金去处
  • 银行贷款的纳税申报表指的所得税还是增值税
  • 维护费可以留到明年抵扣吗?
  • 以前年度企业所得税申报表怎么查
  • 公司与股东的往来款涉税
  • 下岗职工能享受什么待遇
  • 收不回工程款的心情
  • 定期结汇会计分录?
  • 出售使用过的车辆如何交印花税呢
  • 购买电子承兑差额怎么算
  • 收益性支出的项目有哪些
  • PHP:stream_socket_pair()的用法_Stream函数
  • 结存材料实际成本例题
  • 冷门暴利行业
  • PHP:mcrypt_enc_get_key_size()的用法_Mcrypt函数
  • 补缴房产税土地使用税怎么算
  • 储金会是干什么的
  • 企业会计日记账保管期限为()年。A5B10C30D35
  • 夕阳下的大白鲨过山车,美国新泽西州 (© John Van Decker/Alamy)
  • 融资购入的固定资产如何记账
  • 一般纳税人公司出售旧车怎么开票
  • 猿厂猿作设计机构招聘
  • 微信小程序获取地理位置
  • 十大经典排序算法(动图演示C 实现)
  • 外贸公司主要是做什么产品
  • 电影制片过程
  • 商业承兑到期后有效期
  • sql语句中的decode函数
  • 沙子产量表格
  • 国税局领发票是哪些项目需要收取费用
  • 残疾人保障金需要联系专管员吗
  • 分支机构注销后往来账如何处理?
  • 利息收入是一级科目还是
  • 刚成立公司没有发生业务,如何申报企业所得税
  • 支付结算办法实施细则最新
  • 固定资产全套账务处理2021
  • 冲减上月收入会计分录
  • 建安企业增值税预缴
  • 房屋租赁违约金比例
  • 印花税为什么不用计提
  • 费用类科目的明细怎么写
  • 计提附加税减半征收会计分录
  • 管理费用标准制定
  • windows sky high
  • win7电脑开机自启动管理在哪里
  • mssecsvc是什么进程
  • win10预览版和正式版区别
  • Win10 Redstone版Build 14251现身 正式版将在7月份发布
  • jQuery实现非常实用漂亮的select下拉菜单选择效果
  • shell脚本 定义变量
  • android opengl绘图
  • 基于android的app
  • 自动关机的
  • linux Shell获取某目录下所有文件夹的名称
  • window.parent与window.openner区别介绍
  • putty自动退出
  • 浅谈 javascript 事件处理
  • 封装好的中药能带上飞机吗
  • jquery增加一行和删除
  • 电子税务局驱动安装步骤
  • 央企收入归谁
  • 税控盘时钟异常怎么办
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

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

    友情链接: 武汉网站建设