diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-11-10 23:39:50 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-11-10 23:39:50 +0400 |
commit | dcbb40662044aab242223eb5151df2da2552e80d (patch) | |
tree | 1601f354b49603e4a0b50317937a1a1916e358b9 | |
parent | 120ffa94f309b6da0ca9174a1c5fff790ae6071f (diff) |
Fixed bug in distance-based NLL.
-rw-r--r-- | DistNLLCriterion.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/DistNLLCriterion.lua b/DistNLLCriterion.lua index 8f2528a..6f6679f 100644 --- a/DistNLLCriterion.lua +++ b/DistNLLCriterion.lua @@ -42,7 +42,7 @@ end function DistNLLCriterion:denormalize(input) -- denormalize gradients if not self.inputIsLogProbability and not self.inputIsProbability then - self.gradInput = self.inputLogSoftMax:backward(input, self.gradLogInput) + self.gradInput = self.inputLogSoftMax:backward(self.input, self.gradLogInput) elseif not self.inputIsLogProbability then print('TODO: implement nn.Log()') else |