Change Moment to Momentum

rrmina · Mar 31, 2019 · d2bdd65 · d2bdd65
1 parent 2d723e5
commit d2bdd65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eureka/optim.py b/eureka/optim.py
@@ -13,7 +13,7 @@ def step(self):
                 layer.w -= self.lr * layer.dw
                 layer.b -= self.lr * layer.db
 
-class Moment(object):
+class Momentum(object):
     def __init__(self, model_instance, lr=0.01, beta_1=0.9):
         self.model_instance = model_instance
         self.lr = lr