Skip to content

Commit 075bc0f

Browse files
committed
check for right numbers
1 parent cc03eae commit 075bc0f

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

lib/OptimizationOptimisers/src/OptimizationOptimisers.jl

+8-2
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,16 @@ function SciMLBase.__solve(cache::OptimizationCache{
5252
dataiterate = false
5353
end
5454

55-
epochs, maxiters = if isnothing(cache.solver_args.maxiters) ==
55+
epochs, maxiters = if isnothing(cache.solver_args.maxiters) &&
5656
isnothing(cache.solver_args.epochs)
57-
# both of them are `nothing` or have a value
5857
throw(ArgumentError("The number of iterations must be specified with either the epochs or maxiters kwarg. Where maxiters = epochs * length(data)."))
58+
elseif !isnothing(cache.solver_args.maxiters) &&
59+
!isnothing(cache.solver_args.epochs)
60+
if cache.solver_args.maxiters == cache.solver_args.epochs * length(data)
61+
cache.solver_args.epochs, cache.solver_args.maxiters
62+
else
63+
throw(ArgumentError("The number of iterations must be specified with either the epochs or maxiters kwarg. Where maxiters = epochs * length(data)."))
64+
end
5965
elseif isnothing(cache.solver_args.maxiters)
6066
cache.solver_args.epochs, cache.solver_args.epochs * length(data)
6167
elseif isnothing(cache.solver_args.epochs)

0 commit comments

Comments
 (0)