From 6051c37a2c55365a5834d5b4c11d973e2e4e1ad4 Mon Sep 17 00:00:00 2001 From: Determinant Date: Thu, 28 Apr 2016 17:45:21 +0800 Subject: back-propagate error to the second head of nerv.MSELayer --- nerv/layer/mse.lua | 3 +++ 1 file changed, 3 insertions(+) diff --git a/nerv/layer/mse.lua b/nerv/layer/mse.lua index c1ea596..594eb0c 100644 --- a/nerv/layer/mse.lua +++ b/nerv/layer/mse.lua @@ -63,9 +63,12 @@ function MSELayer:back_propagate(bp_err, next_bp_err, input, output, t) t = 1 end local nbe = next_bp_err[1] + local nbe2 = next_bp_err[2] nbe:add(nbe, self.diff[t], 0.0, self.scale) + nbe2:add(nbe2, self.diff[t], 0.0, -self.scale) if bp_err[1] ~= nil then nbe:scale_rows_by_col(bp_err[1]) + nbe2:scale_rows_by_col(bp_err[1]) end end -- cgit v1.2.3