We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 43ce2ab commit 4bf015cCopy full SHA for 4bf015c
layers/modules/l2norm.py
@@ -14,7 +14,7 @@ def __init__(self,n_channels, scale):
14
self.reset_parameters()
15
16
def reset_parameters(self):
17
- init.constant(self.weight,self.gamma)
+ init.constant_(self.weight,self.gamma)
18
19
def forward(self, x):
20
norm = x.pow(2).sum(dim=1, keepdim=True).sqrt()+self.eps
0 commit comments