Remove F.normalize from SoftmaxLoss

vjoki · Jan 29, 2021 · 6b0697b · 6b0697b
1 parent 6f042d8
commit 6b0697b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/snn/librispeech/loss/softmax.py b/snn/librispeech/loss/softmax.py
@@ -14,7 +14,6 @@ def __init__(self, nOut=512, nClasses=251, **kwargs):
     def forward(self, x, label=None):
         # TODO: Not sure what the rationale for N -> nClasses is...
         # B*W*SxN -> B*W*SxnClasses
-        x = F.normalize(x, p=2, dim=1)
         x = self.fc(x)
         nloss = self.criterion(x, label)
         prec1 = accuracy(x.detach(), label.detach(), topk=(1,))[0]