print num iter

Author: pmocz

examples/logo_inverse_problem/logo_inverse_problem.py

@@ -80,6 +80,7 @@ def loss_function(theta, rho_target):
     t0 = time.time()
     sol = opt.run(theta, rho_target)
     print("Inverse-problem solve time (s): ", time.time() - t0)
+    print("number of iterations: ", sol.state.iter_num)
     theta = jnp.mod(sol.params, 2.0 * jnp.pi) - jnp.pi
     print("Mean theta:", jnp.mean(theta))