reduce maxiters

SciML · Nov 20, 2023 · 1d79acf · 1d79acf
1 parent b59a625
commit 1d79acf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/docs/src/examples/normalizing_flows.md b/docs/src/examples/normalizing_flows.md
@@ -37,7 +37,7 @@ adtype = Optimization.AutoForwardDiff()
 optf = Optimization.OptimizationFunction((x, p) -> loss(x), adtype)
 optprob = Optimization.OptimizationProblem(optf, ps)
 
-res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 100, callback = cb)
+res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 20, callback = cb)
 
 optprob2 = Optimization.OptimizationProblem(optf, res1.u)
 res2 = Optimization.solve(optprob2, Optim.LBFGS(); allow_f_increases = false,
@@ -113,8 +113,7 @@ adtype = Optimization.AutoForwardDiff()
 optf = Optimization.OptimizationFunction((x, p) -> loss(x), adtype)
 optprob = Optimization.OptimizationProblem(optf, ps)
 
-res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 100,
-    callback = cb)
+res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 20, callback = cb)
 ```
 
 We then complete the training using a different optimizer, starting from where `Adam` stopped.