Skip to content

Commit 1d79acf

Browse files
committed
reduce maxiters
1 parent b59a625 commit 1d79acf

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

docs/src/examples/normalizing_flows.md

+2-3
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ adtype = Optimization.AutoForwardDiff()
3737
optf = Optimization.OptimizationFunction((x, p) -> loss(x), adtype)
3838
optprob = Optimization.OptimizationProblem(optf, ps)
3939
40-
res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 100, callback = cb)
40+
res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 20, callback = cb)
4141
4242
optprob2 = Optimization.OptimizationProblem(optf, res1.u)
4343
res2 = Optimization.solve(optprob2, Optim.LBFGS(); allow_f_increases = false,
@@ -113,8 +113,7 @@ adtype = Optimization.AutoForwardDiff()
113113
optf = Optimization.OptimizationFunction((x, p) -> loss(x), adtype)
114114
optprob = Optimization.OptimizationProblem(optf, ps)
115115
116-
res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 100,
117-
callback = cb)
116+
res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 20, callback = cb)
118117
```
119118

120119
We then complete the training using a different optimizer, starting from where `Adam` stopped.

0 commit comments

Comments
 (0)