Pytorch卷积层手动初始化权值的实例
更新时间：2019年08月17日 11:59:24 作者：Learn2Learn
今天小编就为大家分享一篇Pytorch卷积层手动初始化权值的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
由于研究关系需要自己手动给卷积层初始化权值，但是好像博客上提到的相关文章比较少（大部分都只提到使用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卷积层手动初始化权值的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
我喜欢看电影，可以说大部分热门电影我都看过。处理爱好的目的，我看了看豆瓣热映的电影列表。于是我写了这个爬虫把豆瓣热映的电影都爬了下来。对页面的处理主要是需要点击显示全部电影，然后爬取影片属性，最后输出文本。采用的还是scrapy框架。顺便聊聊我的实现过程吧2021-11-11
今天小编就为大家分享一篇django admin.py 外键,反向查询的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
今天小编就为大家分享一篇解决tensorflow添加ptb库的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
导入文件目的就是为了执行文件,下面这篇文章主要给大家介绍了关于Python中模块或py文件导入的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2023-04-04
这篇文章主要介绍了Python数据类型之列表和元组的相关知识，列表是一组有序项目的集合 ,可变的数据类型可 进行增删改查，本文通过实例文字相结合的形式给大家介绍的非常详细 ,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了python二分法的实现代码，大家可以参考使用2013-11-11
这篇文章主要为大家详细介绍了pycharm实现猜数游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-12-12
本文介绍如何使用Python动态监控程序日志的内容，这里的动态指的是日志文件不断的追加新的日志内容，动态监控是指监控日志新追加的日志内容2014-02-02
这篇文章主要介绍了Python面向对象思想与应用,较为详细的分析了Python面向对象思想与原理,并结合实例形式分析了类与对象相关定义、用法及操作注意事项,需要的朋友可以参考下2019-04-04
caffe的python接口deploy生成caffemodel分类新的图片
这篇文章主要为大家介绍了caffe的python接口生成deploy文件学习以及用训练好的模型（caffemodel）来分类新的图片示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-06-06
最新评论