diff options
author | Determinant <[email protected]> | 2015-06-26 01:03:54 +0800 |
---|---|---|
committer | Determinant <[email protected]> | 2015-06-26 01:03:54 +0800 |
commit | bc17acd4c5f98df4e00b7c85e479cbff2d5da5a7 (patch) | |
tree | b05de64bc93d13c23ff11aa7f6650ad9f4bd8dc1 /nerv/layer | |
parent | aae4195c3898c0da0be5aae0b80e633185e1e242 (diff) |
...
Diffstat (limited to 'nerv/layer')
-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 |