diff options
| author | cloudygoose <cloudygooseg@gmail.com> | 2015-06-26 00:02:57 +0800 |
|---|---|---|
| committer | cloudygoose <cloudygooseg@gmail.com> | 2015-06-26 00:02:57 +0800 |
| commit | d5e95b665b9dc1890fb575fcc9fca5087eaeb674 (patch) | |
| tree | 67512bdf955e2753547ebdac2e18f837ce6fa5b0 | |
| parent | d7e14b19816732d8efe6b586aaa493c2d924124c (diff) | |
softmax_ce colsum adapt to matrix change, using [0][0]
| -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 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 |