TrainingPhase

struct defined in module FluxTraining.Phases


			TrainingPhase() <: AbstractTrainingPhase

A regular training phase for supervised learning. It iterates over batches in learner.data.training and updates the model parameters using learner.optim after calculating the gradients.

Throws the following events in this order:

It writes the following step state to learner.state, grouped by the event from which on it is available.

  • StepBegin:

    • xs and ys: encoded input and target (batch)

  • LossBegin:

    • ŷs: model output

  • BackwardBegin:

    • loss: loss

  • BackwardEnd:

    • grads: calculated gradients