File tree 1 file changed +2
-3
lines changed
1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -37,7 +37,7 @@ adtype = Optimization.AutoForwardDiff()
37
37
optf = Optimization.OptimizationFunction((x, p) -> loss(x), adtype)
38
38
optprob = Optimization.OptimizationProblem(optf, ps)
39
39
40
- res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 100 , callback = cb)
40
+ res1 = Optimization.solve(optprob, Adam(0.1); maxiters = 20 , callback = cb)
41
41
42
42
optprob2 = Optimization.OptimizationProblem(optf, res1.u)
43
43
res2 = Optimization.solve(optprob2, Optim.LBFGS(); allow_f_increases = false,
@@ -113,8 +113,7 @@ adtype = Optimization.AutoForwardDiff()
113
113
optf = Optimization.OptimizationFunction((x, p) -> loss(x), adtype)
114
114
optprob = Optimization.OptimizationProblem(optf, ps)
115
115
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)
118
117
```
119
118
120
119
We then complete the training using a different optimizer, starting from where ` Adam ` stopped.
You can’t perform that action at this time.
0 commit comments