fix a bug: input[1] should be input[2] (since Lua arrays are 1-based)

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