-
-
Notifications
You must be signed in to change notification settings - Fork 11
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ReduceLRonPlateau support #11
Comments
So that type of schedule will not fit into the iteration interface, because it is inherently stateful (depends on more than the iteration index). The closest example of a schedule that fits that style is If we do want to include lots of stateful schedules, then maybe we should think about what that means. Another example of a stateful "schedule" is Specifically for plateau = Flux.early_stopping(validation, patience = 5)
schedule = ParameterSchedulers.Stateful(Exp(λ = 0.1, γ = 0.5))
opt = Descent()
for epoch in 1:nepochs
if plateau()
opt.eta = next!(schedule)
end
# ...
end So, a generic implementation would pull the idea of a predicate into |
I wasn't sure how best to include such schedulers with the current API (especially with
iterate
). Idea is same as PyTorch ReduceLROnPlateau.The text was updated successfully, but these errors were encountered: