Update OptimizationOptimisers.jl

Modify check for last iteration to set the parameters to the best minimizer if save_best if enabled.

Author: lsmohan

lib/OptimizationOptimisers/src/OptimizationOptimisers.jl

@@ -130,12 +130,12 @@ function SciMLBase.__solve(cache::OptimizationCache{
                         min_err = x
                         min_θ = copy(θ)
                     end
-                    if i == length(data)*epochs  #Last iter, revert to best.
+                    if iterations == length(data)*epochs  #Last iter, revert to best.
                         opt = min_opt
                         x = min_err
                         θ = min_θ
                         cache.f.grad(G, θ, d)
-                        opt_state = Optimization.OptimizationState(iter = i,
+                        opt_state = Optimization.OptimizationState(iter = iterations,
                             u = θ,
                             objective = x[1],
                             grad = G,