Pytorch 快速参数权重初始化-CSDN博客
阿里云国内75折 回扣 微信号:monov8 |
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6 |
定义一个函数
这里比如要初始化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