pytorch 预训练层的使用方法
更新时间：2019年08月20日 09:38:37 作者：zzw小凡
今天小编就为大家分享一篇pytorch 预训练层的使用方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch 预训练层的使用方法
将其他地方训练好的网络，用到新的网络里面
加载预训练网络
1.原先已经训练好一个网络 AutoEncoder_FC()
2.首先加载该网络，读取其存储的参数
3.设置一个参数集
cnnpre = AutoEncoder_FC()
cnnpre.load_state_dict(torch.load('autoencoder_FC.pkl')['state_dict'])
cnnpre_dict =cnnpre.state_dict()
加载新网络
1.设置新的网络
2.设置新网络参数集
cnn= AutoEncoder() cnn_dict = cnn.state_dict()
更新新网络参数
1.将两个参数集比对，存在的网络参数保留
2.使用保留下的参数更新新网络参数集
3.加载新网络参数集到新网络中
cnnpre_dict = {k: v for k, v in cnnpre_dict.items() if k in cnn_dict}
cnn_dict.update(cnnpre_dict)
cnn.load_state_dict(cnn_dict)
以上这篇pytorch 预训练层的使用方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要给大家介绍了关于Python如何利用正则表达式爬取爬取古诗文网站信息，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-12-12
今天小编就为大家分享一篇利用Python读取txt文档的方法讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
这篇文章主要介绍了详解TensorFlow在windows上安装与简单示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-03-03
今天小编就为大家分享一篇在Django中URL正则表达式匹配的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
尽管网上搜到许多篇关于pycharm如何安装pygame库,但是跟着要求做还是安装失败,下面这篇文章主要给大家介绍了关于python中pygame安装不成功的解决办法,需要的朋友可以参考下2023-05-05
这篇文章主要介绍了Python入门教程(十五)Python的字典,Python是一门非常强大好用的语言,也有着易上手的特性,本文为入门教程,需要的朋友可以参考下2023-04-04
这篇文章主要介绍了教你使用Python提取视频中的美女图片,使用 Python 写一个逐帧无损保存视频画面的小脚本，大概分为三个步骤，详细过程跟随小编一起看看吧2022-04-04
Linux下通过python访问MySQL、Oracle、SQL Server数据库的方法
这篇文章主要介绍了Linux下通过python访问MySQL、Oracle、SQL Server数据库的方法,需要的朋友可以参考下2016-04-04
这篇文章主要为大家介绍了python利用opencv调用摄像头实现目标检测的示例代码，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-05-05
这篇文章主要介绍了如何使用 wxPython 模块和 python-pptx 模块来编写一个程序，用于生成包含首页、内容页和感谢页的 PPTX 文档，感兴趣的小伙伴可以学习一下2023-08-08
最新评论