From d5e95b665b9dc1890fb575fcc9fca5087eaeb674 Mon Sep 17 00:00:00 2001 From: cloudygoose Date: Fri, 26 Jun 2015 00:02:57 +0800 Subject: softmax_ce colsum adapt to matrix change, using [0][0] --- nerv/layer/softmax_ce.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 -- cgit v1.2.3