aboutsummaryrefslogtreecommitdiff
path: root/layer
diff options
context:
space:
mode:
Diffstat (limited to 'layer')
-rw-r--r--layer/softmax_ce.lua4
1 files changed, 3 insertions, 1 deletions
diff --git a/layer/softmax_ce.lua b/layer/softmax_ce.lua
index 3dfebc5..09eb3a9 100644
--- a/layer/softmax_ce.lua
+++ b/layer/softmax_ce.lua
@@ -27,6 +27,8 @@ function SoftmaxCELayer:propagate(input, output)
local ce = soutput:create()
ce:log_elem(soutput)
ce:mul_elem(ce, input[2])
+-- print(input[1][0])
+-- print(soutput[1][0])
-- add total ce
self.total_ce = self.total_ce - ce:rowsum():colsum()[0]
self.total_frames = self.total_frames + soutput:nrow()
@@ -34,5 +36,5 @@ end
function SoftmaxCELayer:back_propagate(next_bp_err, bp_err, input, output)
-- softmax output - label
- next_bp_err[1]:add(self.soutput, input[1], 1.0, -1.0)
+ next_bp_err[1]:add(self.soutput, input[2], 1.0, -1.0)
end