Add callback support for Optimisers.jl

FluxML / FluxTraining.jl

A flexible neural net training library inspired by fast.ai

https://fluxml.ai/FluxTraining.jl

MIT License

117 stars 25 forks source link

Closed lorenzoh closed 2 years ago

lorenzoh commented 2 years ago

[x] Allow ToDevice read learner.optimizer so it can dispatch on the optimiser type
[x] Make hyperparameter scheduling work