aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorcloudygoose <[email protected]>2015-06-26 00:02:57 +0800
committercloudygoose <[email protected]>2015-06-26 00:02:57 +0800
commitd5e95b665b9dc1890fb575fcc9fca5087eaeb674 (patch)
tree67512bdf955e2753547ebdac2e18f837ce6fa5b0
parentd7e14b19816732d8efe6b586aaa493c2d924124c (diff)
softmax_ce colsum adapt to matrix change, using [0][0]
-rw-r--r--nerv/layer/softmax_ce.lua2
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/layer/softmax_ce.lua b/nerv/layer/softmax_ce.lua
index daf891e..c78d462 100644
--- a/nerv/layer/softmax_ce.lua
+++ b/nerv/layer/softmax_ce.lua
@@ -42,7 +42,7 @@ function SoftmaxCELayer:propagate(input, output)
output[1]:copy_fromd(ce)
end
-- add total ce
- self.total_ce = self.total_ce - ce:colsum()[0]
+ self.total_ce = self.total_ce - ce:colsum()[0][0]
self.total_frames = self.total_frames + softmax:nrow()
-- TODO: add colsame for uncompressed label
if self.compressed then