A Julia package for using and writing powerful, extensible training loops for deep learning models.
Install like any other Julia package using the package manager:
]add FluxTraining
After installation, import it, create a Learner
from a Flux.jl model, data iterators, an optimizer, and a loss function. Finally train with fit!
.
using FluxTraining
learner = Learner(model, lossfn)
fit!(learner, 10, (trainiter, validiter))
The design of FluxTraining.jl's two-way callbacks is adapted from fastai's training loop.