Skip to content

Commit d351239

Browse files
committed
fix legacy ClassNLLCriterion for upstream change
1 parent 1b1579c commit d351239

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torch/legacy/nn/ClassNLLCriterion.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@ def updateOutput(self, input, target):
2424
self.output_tensor,
2525
self.sizeAverage,
2626
self.weights,
27-
self.total_weight_tensor
27+
self.total_weight_tensor,
28+
-100
2829
)
2930
self.output = self.output_tensor[0]
3031
return self.output
@@ -40,7 +41,8 @@ def updateGradInput(self, input, target):
4041
self.gradInput,
4142
self.sizeAverage,
4243
self.weights,
43-
self.total_weight_tensor
44+
self.total_weight_tensor,
45+
-100
4446
)
4547

4648
return self.gradInput

0 commit comments

Comments
 (0)