PyTorch 参数初始化

来源:互联网 发布:小米网络机顶盒哪个好 编辑:程序博客网 时间:2024/06/06 20:13

import torch.nn.init as init

self.conv1 = nn.Conv2d(3, 20, 5, stride=1, bias=True) 
init.xavier_uniform(self.conv1.weight, gain=np.sqrt(2.0))
init.constant(self.conv1.bias, 0.1)

0 0
原创粉丝点击