diff options
Diffstat (limited to 'nerv/layer/softmax_ce.lua')
| -rw-r--r-- | nerv/layer/softmax_ce.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/layer/softmax_ce.lua b/nerv/layer/softmax_ce.lua index c78d462..f878a2f 100644 --- a/nerv/layer/softmax_ce.lua +++ b/nerv/layer/softmax_ce.lua @@ -46,7 +46,7 @@ function SoftmaxCELayer:propagate(input, output) self.total_frames = self.total_frames + softmax:nrow() -- TODO: add colsame for uncompressed label if self.compressed then - self.total_correct = self.total_correct + classified:colsame(input[2])[0] + self.total_correct = self.total_correct + classified:colsame(input[2])[0][0] end end |