pytorch 使用预训练层
来源:互联网 发布:什么是软件危机 编辑:程序博客网 时间:2024/06/06 02:52
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)
阅读全文
0 0
- pytorch 使用预训练层
- PyTorch预训练
- PyTorch预训练
- 使用pytorch预训练模型分类与特征提取
- pytorch使用(四)训练网络
- pytorch 如何加载部分预训练模型
- pytorch训练imagenet分类
- PyTorch学习系列(十)——如何在训练时固定一些层?
- pytorch 使用
- 【pytorch】训练集的读取
- PyTorch代码学习-ImageNET训练
- PyTorch(三)——使用训练好的模型测试自己图片
- PyTorch学习系列(十六)——如何使用cuda进行训练?
- 什么是PyTorch,为何要使用PyTorch
- PyTorch学习系列(十五)——如何加载预训练模型?
- pytorch学习笔记(十一):fine-tune 预训练的模型
- PyTorch批训练及优化器比较
- PyTorch 设置使用GPU
- spring中基于BeanPostProcessor动态替换指定的bean
- linux下find和grep的区别
- centos6上超详细LAMP编译安装过程以及所遇到的问题
- 游戏中的战争迷雾
- ASP.NET MVC
- pytorch 使用预训练层
- 小程序.silk转MP3格式
- [转]【OpenCV入门教程之十二】OpenCV边缘检测:Canny算子,Sobel算子,Laplace算子,Scharr滤波器合辑
- Python自动化运维 技术与最佳实践
- Baidu地图
- scala基础
- 二进制文件与base64编码文本文件转换
- ASP.NET MVC
- 牧场上的草泥马(游荡的奶牛)