fix bounds translation

fixes #498
SciML · Mar 4, 2021 · 331ec8b · 331ec8b
1 parent 56f2102
commit 331ec8b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/train.jl b/src/train.jl
@@ -1,6 +1,7 @@
-function sciml_train(loss, θ, opt, adtype::DiffEqBase.AbstractADType = GalacticOptim.AutoZygote(), args...; kwargs...)
+function sciml_train(loss, θ, opt, adtype::DiffEqBase.AbstractADType = GalacticOptim.AutoZygote(), args...;
+                     lower_bounds = nothing, upper_bounds = nothing, kwargs...)
   optf = GalacticOptim.OptimizationFunction((x, p) -> loss(x), adtype)
   optfunc = GalacticOptim.instantiate_function(optf, θ, adtype, nothing)
-  optprob = GalacticOptim.OptimizationProblem(optfunc, θ; kwargs...)
+  optprob = GalacticOptim.OptimizationProblem(optfunc, θ; lb = lower_bounds, ub = upper_bounds, kwargs...)
   GalacticOptim.solve(optprob, opt, args...; kwargs...)
 end