diff options
Diffstat (limited to 'nerv/layer/affine_recurrent.lua')
| -rw-r--r-- | nerv/layer/affine_recurrent.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/nerv/layer/affine_recurrent.lua b/nerv/layer/affine_recurrent.lua index c8cd382..b465e95 100644 --- a/nerv/layer/affine_recurrent.lua +++ b/nerv/layer/affine_recurrent.lua @@ -61,8 +61,8 @@ function Recurrent:update(bp_err, input, output) bp:add(bp, bp_err[1]:colsum(), 1.0-gconf.lrate*gconf.wcost/gconf.batch_size, -gconf.lrate/gconf.batch_size) end else - self.ltp_hh_grad:mul(input[2], bp_err[1], 1.0, 0.0, 'T', 'N') - self.ltp_hh:update(self.ltp_hh_grad) + --self.ltp_hh_grad:mul(input[2], bp_err[1], 1.0, 0.0, 'T', 'N') + self.ltp_hh:updateEI(bp_err[1], input[2]) self.bp:update(bp_err[1]:colsum()) end end |