Getting started

Let's look at a simple training example. In FluxTraining.jl, a Learner holds all state necessary for training. To get started, you need

• a model
• training and validation data iterators
• a loss function; and
• an optimizer

First we define the necessary pieces:

model = ...
traindata, valdata = ...
lossfn = Flux.Losses.mse
opt = Flux.ADAM(0.01)

Then we construct a Learner:

learner = Learner(model, lossfn; data=(traindata, valdata))

And train for 10 epochs:

fit!(learner, 10)