Pytorch卷积层手动初始化权值的实例


Posted in Python onAugust 17, 2019

由于研究关系需要自己手动给卷积层初始化权值,但是好像博客上提到的相关文章比较少(大部分都只提到使用nn.init里的按照一定分布初始化方法),自己参考了下Pytorch的官方文档,发现有两种方法吧。

所以mark下。

import torch
import torch.nn as nn
import torch.optim as optim
import numpy as np

# 第一一个卷积层,我们可以看到它的权值是随机初始化的
w=torch.nn.Conv2d(2,2,3,padding=1)
print(w.weight)


# 第一种方法
print("1.使用另一个Conv层的权值")
q=torch.nn.Conv2d(2,2,3,padding=1) # 假设q代表一个训练好的卷积层
print(q.weight) # 可以看到q的权重和w是不同的
w.weight=q.weight # 把一个Conv层的权重赋值给另一个Conv层
print(w.weight)

# 第二种方法
print("2.使用来自Tensor的权值")
ones=torch.Tensor(np.ones([2,2,3,3])) # 先创建一个自定义权值的Tensor,这里为了方便将所有权值设为1
w.weight=torch.nn.Parameter(ones) # 把Tensor的值作为权值赋值给Conv层,这里需要先转为torch.nn.Parameter类型,否则将报错
print(w.weight)

效果预览

Pytorch卷积层手动初始化权值的实例

以上这篇Pytorch卷积层手动初始化权值的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
go和python调用其它程序并得到程序输出
Feb 10 Python
python实现超简单端口转发的方法
Mar 13 Python
Python命令行解析模块详解
Feb 01 Python
python实现SOM算法
Feb 23 Python
火车票抢票python代码公开揭秘!
Mar 08 Python
利用Anaconda简单安装scrapy框架的方法
Jun 13 Python
对Pyhon实现静态变量全局变量的方法详解
Jan 11 Python
Python制作词云图代码实例
Sep 09 Python
python 实现一个反向单位矩阵示例
Nov 29 Python
在TensorFlow中屏蔽warning的方式
Feb 04 Python
pycharm中import呈现灰色原因的解决方法
Mar 04 Python
使用Python判断一个文件是否被占用的方法教程
Dec 16 Python
pytorch自定义初始化权重的方法
Aug 17 #Python
在Pytorch中使用样本权重(sample_weight)的正确方法
Aug 17 #Python
获取Pytorch中间某一层权重或者特征的例子
Aug 17 #Python
pyenv与virtualenv安装实现python多版本多项目管理
Aug 17 #Python
pytorch 获取层权重,对特定层注入hook, 提取中间层输出的方法
Aug 17 #Python
关于PyTorch源码解读之torchvision.models
Aug 17 #Python
django项目用higcharts统计最近七天文章点击量
Aug 17 #Python
You might like
laravel5.4生成验证码的实例讲解
2017/08/05 PHP
JS宝典学习笔记(下)
2007/01/10 Javascript
getElementById在任意一款浏览器中都可以用吗的疑问回复
2007/05/13 Javascript
JavaScript 语法集锦 脚本之家基础推荐
2009/11/15 Javascript
jquery中的mouseleave和mouseout的区别 模仿下拉框效果
2012/02/07 Javascript
node.js中的fs.rename方法使用说明
2014/12/16 Javascript
jquery带翻页动画的电子杂志代码分享
2015/08/21 Javascript
基于jQuery实现的扇形定时器附源码下载
2015/10/20 Javascript
浅析Bootstrap组件之面板组件
2016/05/04 Javascript
AngularJS基础 ng-keyup 指令简单示例
2016/08/02 Javascript
BootStrap glyphicon图标无法显示的解决方法
2016/09/06 Javascript
JS实现的简单图片切换功能示例【测试可用】
2017/02/14 Javascript
浅谈React深度编程之受控组件与非受控组件
2017/12/26 Javascript
js提取中文拼音首字母的封装工具类
2018/03/12 Javascript
JS实现的input选择图片本地预览功能示例
2018/08/29 Javascript
解决vue项目nginx部署到非根目录下刷新空白的问题
2018/09/27 Javascript
vue 实现 rem 布局或vw 布局的方法
2019/11/13 Javascript
如何解决vue在ios微信"复制链接"功能问题
2020/03/26 Javascript
JS绘图Flot如何实现动态可刷新曲线图
2020/10/16 Javascript
Python中的闭包详细介绍和实例
2014/11/21 Python
python类和继承用法实例
2015/07/07 Python
详解Django-channels 实现WebSocket实例
2019/08/22 Python
python动态视频下载器的实现方法
2019/09/16 Python
Python FFT合成波形的实例
2019/12/04 Python
基于Python fminunc 的替代方法
2020/02/29 Python
JetBrains PyCharm(Community版本)的下载、安装和初步使用图文教程详解
2020/03/19 Python
翻转数列python实现,求前n项和,并能输出整个数列的案例
2020/05/03 Python
Python Pandas 对列/行进行选择,增加,删除操作
2020/05/17 Python
CSS3实现王者荣耀匹配人员加载页面的方法
2019/04/16 HTML / CSS
三只松鼠官方旗舰店:全网坚果销售第1
2017/11/25 全球购物
阿迪达斯荷兰官方网站:adidas荷兰
2018/03/16 全球购物
中秋节国旗下演讲稿
2014/09/13 职场文书
四风问题自我剖析材料
2014/10/07 职场文书
产品质量保证书范本
2015/02/27 职场文书
2015年信访工作总结
2015/04/07 职场文书
Python机器学习之基础概述
2021/05/19 Python