Update loss argument, fix warning (pytorch#223)

vcarpani · Aug 3, 2018 · ffe9b6e · ffe9b6e
1 parent c4563dd
commit ffe9b6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ignite/metrics/test_loss.py b/tests/ignite/metrics/test_loss.py
@@ -42,7 +42,7 @@ def test_compute_on_criterion():
 
 
 def test_non_averaging_loss():
-    loss = Loss(nn.NLLLoss(reduce=False))
+    loss = Loss(nn.NLLLoss(reduction='none'))
 
     y_pred = torch.Tensor([[0.1, 0.4, 0.5], [0.1, 0.7, 0.2]]).log()
     y = torch.LongTensor([2, 2])