0.4.4 fix bug

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = CGDs
-version = 0.4.3
+version = 0.4.4
 author = Hongkai Zheng
 author_email = [email protected]
 description = Adaptive Competitive Gradient Descent optimizer

src/CGDs/acgd.py

@@ -52,7 +52,7 @@ def set_lr(self, lr_max, lr_min):
         print('Maximizing side learning rate: {:.4f}\n '
               'Minimizing side learning rate: {:.4f}'.format(lr_max, lr_min))
 
-    def step(self, loss, trigger=None):
+    def step(self, loss, trigger=0.0):
         '''
         Update model weights using ACGD
         :param loss: objective function

src/CGDs/cgd_utils.py

@@ -82,7 +82,7 @@ def conjugate_gradient(grad_x, grad_y,
 
 def Hvp_vec(grad_vec, params, vec,
             retain_graph=False,
-            trigger=None,
+            trigger=0.0,
             reducer=None,
             rebuild=False):
     '''