当前位置:首页 > 编程笔记 > 正文
已解决

Pytorch 快速参数权重初始化

来自网友在路上 11058105提问 提问时间:2023-11-10 01:49:47阅读次数: 105

最佳答案 问答题库1058位专家为你答疑解惑

定义一个函数:

这里比如要初始化2维卷积权重值,采用xaiver 数据分布,还有很多其他的数据分布可以探索

def weights_init(m):if isinstance(m, nn.Conv2d):xavier(m.weight.data)xavier(m.bias.data)

然后定义一个含2维卷积的网络,并对该网络中的2维卷积层权重进行初始化操作。

net = Residual() # generate an instance network from the Net class
net.apply(weights_init) # apply weight init

查看全文

99%的人还看了

猜你感兴趣

版权申明

本文"Pytorch 快速参数权重初始化":http://eshow365.cn/6-36750-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!