TrainingRunner

class TrainingRunner<T : DType, V>(val model: Module<T, V>, val loss: Loss, val optimizer: Optimizer)(source)

Runner that holds training components and provides methods to execute training steps.

Constructors

constructor(model: Module<T, V>, loss: Loss, optimizer: Optimizer)

Perform a single training step: forward, backward, optimizer step, and zero grad.

Optional helper to run a training loop over a dataset.