- A+
所属分类:
PyTorch
torch.nn.init
torch.nn.init.calculate_gain(nonlinearity,param=None)
对于给定的非线性函数,返回推荐的增益值。这些值如下所示:
nonlinearity | gain |
---|---|
linear | 1 |
conv{1,2,3}d | 1 |
sigmoid | 1 |
tanh | 5/3 |
relu | sqrt(2) |
leaky_relu | sqrt(2/(1+negative_slope^2)) |