added __slots__ to SGD with Momentum

RoboticsClubIITJ · Mar 13, 2021 · cd6a034 · cd6a034
1 parent 0de7738
commit cd6a034
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/MLlib/optim/sgd.py b/MLlib/optim/sgd.py
@@ -49,6 +49,8 @@ class SGDWithMomentum(optim.Optimizer):
               The momentum (beta)
     """
 
+    __slots__ = ('beta', 'lr', 'v')
+
     def __init__(self, parameters, lr=0.001, momentum=0.9):
         super().__init__(parameters)
         self.beta = momentum