change updateEI to update_by_err_input

1 files changed, 4 insertions, 4 deletions

diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua
index c5084c4..3ba9408 100644
--- a/nerv/layer/affine.lua
+++ b/nerv/layer/affine.lua
@@ -17,7 +17,7 @@ function MatrixParam:train_init()
     self.correction:fill(0)
 end
 
-function MatrixParam:update(gradient)
+function MatrixParam:update_by_gradient(gradient)
     local gconf = self.gconf
     if gconf.momentum > 0 then
         self.correction:add(self.correction, gradient, gconf.momentum, 1.0)
@@ -31,7 +31,7 @@ function MatrixParam:update(gradient)
     end
 end
 
-function MatrixParam:updateEI(err, input)
+function MatrixParam:update_by_err_input(err, input)
     local gconf = self.gconf
     if gconf.momentum > 0 then
         self.correction:mul(input, err, 1.0, gconf.momentum, 'T', 'N')
@@ -108,8 +108,8 @@ function AffineLayer:update(bp_err, input, output)
             self.bp.trans:add(self.bp.trans, bp_err[1]:colsum(), 1.0 - gconf.lrate * gconf.wcost / gconf.batch_size, - gconf.lrate / gconf.batch_size)
         end
     else
-        self.ltp:updateEI(bp_err[1], input[1])
-        self.bp:update(bp_err[1]:colsum())
+        self.ltp:update_by_err_input(bp_err[1], input[1])
+        self.bp:update_by_gradient(bp_err[1]:colsum())
     end
 end