diff options
author | Adam Paszke <adam.paszke@gmail.com> | 2016-04-26 19:38:28 +0300 |
---|---|---|
committer | Adam Paszke <adam.paszke@gmail.com> | 2016-04-26 22:23:50 +0300 |
commit | 05fae45d94793f30e9e02a1b640be8b8bbc9b0d9 (patch) | |
tree | ba3ee86dc454534090ca740851ef3cf84f21c0f4 /ClassNLLCriterion.lua | |
parent | 7715d0b86f386b478f75b0bd5d4aaa604a9a0681 (diff) |
Fix ClassNLLCriterion buffer
Diffstat (limited to 'ClassNLLCriterion.lua')
-rw-r--r-- | ClassNLLCriterion.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ClassNLLCriterion.lua b/ClassNLLCriterion.lua index d704ee1..8e8acbf 100644 --- a/ClassNLLCriterion.lua +++ b/ClassNLLCriterion.lua @@ -44,7 +44,7 @@ function ClassNLLCriterion:updateOutput(input, target) self.output_tensor:cdata(), self.sizeAverage, THNN.optionalTensor(self.weights), - THNN.optionalTensor(self.total_weight_tensor) + self.total_weight_tensor:cdata() ) self.output = self.output_tensor[1] return self.output, self.total_weight_tensor[1] @@ -67,7 +67,7 @@ function ClassNLLCriterion:updateGradInput(input, target) self.gradInput:cdata(), self.sizeAverage, THNN.optionalTensor(self.weights), - THNN.optionalTensor(self.total_weight_tensor) + self.total_weight_tensor:cdata() ) return self.gradInput |