Interface to keras SGD

q-optimize · Jan 25, 2024 · c2fa787 · c2fa787
1 parent 23acf38
commit c2fa787
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/c3/libraries/algorithms.py b/c3/libraries/algorithms.py
@@ -284,9 +284,9 @@ def tf_fun():
 
     opt_sgd = tf.keras.optimizers.SGD(learning_rate=learning_rate, momentum=momentum)
 
-    for _ in range(iters):
-        step_count = opt_sgd.minimize(tf_fun, [var])
-        print(f"epoch {step_count.numpy()}: func_value: {tf_fun()}")
+    for ii in range(iters):
+        opt_sgd.minimize(tf_fun, [var])
+        print(f"iter {ii}: func_value: {tf_fun()}")
 
     result = OptimizeResult(x=var.numpy(), success=True)
     return result