0.4.5 fix the bug in the case of nstep=1 in CG

devzhk · Jun 7, 2022 · 5892a6e · 5892a6e
1 parent 4bda067
commit 5892a6e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = CGDs
-version = 0.4.4
+version = 0.4.5
 author = Hongkai Zheng
 author_email = [email protected]
 description = Adaptive Competitive Gradient Descent optimizer

diff --git a/src/CGDs/cgd_utils.py b/src/CGDs/cgd_utils.py
@@ -165,7 +165,6 @@ def general_conjugate_gradient(grad_x, grad_y,
                      rebuild=rebuild).mul_(lr_x)
         Avx = x + h2
         r = b.clone() - Avx
-        nsteps -= 1
 
     if grad_x.shape != b.shape:
         raise RuntimeError('CG: hessian vector product shape mismatch')