I am happy to announce the release of FluxTraining.jl, a deep learning training package for Flux models.
It has an extensible callback system inspired by fastai’s and comes with features like metrics, hyperparameter scheduling, TensorBoard logging and model checkpointing. It also makes it easy to customize the training loop.
You can find the documentation on all those features here.
Let me know if you find it useful or something is unclear, otherwise, happy training!