FluxTraining.runstep
function defined in module
FluxTraining
runstep(stepfn, learner, phase) -> state
Run
stepfn
inside the context of a step. Calls
stepfn(handle, state)
where
handle(e)
can be called to dispatch events and
state
is a
PropDict
which step data, gradients and losses can be written to. Return
state
.
Takes care of dispatching
StepBegin
and
StepEnd
events as well as handling
CancelStepException
s.
There are
2
methods for runstep:
runstep(stepfn, learner, phase::FluxTraining.Phases.Phase)runstep(stepfn, learner, phase::FluxTraining.Phases.Phase, initialstate)