kaiming_normal

【torch.nn.init】初始化参数方法解读

文章目录torch.nn.init均匀分布正态分布常数分布全1分布全0分布对角分布dirac分布xavier_uniform分布xavier_normal分布kaiming_uniform分布kaiming_normal
zyw2002·2023-08-20 13:27

pytorch权重初始化之kaiming_normal初始化

importtorchimporttorch.nnasnnw=torch.empty(2,3)#torch.nn.init.kaiming_normal_(tensor,a=0,mode='fan_in',nonlinearity='leaky_relu')nn.init.kaiming_normal_(w,mode='fan_out',nonlinearity='relu')#tensor([[
是鲤鱼呀·2020-08-13 17:49
上一页 1 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他