diff options
author | nicholas-leonard <nick@nikopia.org> | 2014-07-09 02:44:40 +0400 |
---|---|---|
committer | nicholas-leonard <nick@nikopia.org> | 2014-07-09 02:44:40 +0400 |
commit | e4c8470dfd45985e67a4d170ea965fc69daeabe2 (patch) | |
tree | ffa918806954d399204b1c5096b33ff472b067c3 /SoftMaxTree.lua | |
parent | 037c8cdbac310be57ffb921492daf286806020bc (diff) |
bug
Diffstat (limited to 'SoftMaxTree.lua')
-rw-r--r-- | SoftMaxTree.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/SoftMaxTree.lua b/SoftMaxTree.lua index d074063..6b783a0 100644 --- a/SoftMaxTree.lua +++ b/SoftMaxTree.lua @@ -171,8 +171,9 @@ end function SoftMaxTree:updateGradInput(inputTable, gradOutput) local input, target = unpack(inputTable) if self.gradInput then - return input.nn.SoftMaxTree_updateGradInput(self, input, gradOutput, target) + input.nn.SoftMaxTree_updateGradInput(self, input, gradOutput, target) end + return self.gradInput end function SoftMaxTree:accGradParameters(inputTable, gradOutput, scale) |