Skip to main content

Module train

Module train 

Source
Expand description

High-level training and evaluation APIs.

These methods validate dataset shapes and return Result, while internally using allocation-free per-sample forward/backward passes.

Structs§

EpochReport
Report for a single epoch.
EvalReport
Output of Mlp::evaluate.
FitConfig
Configuration for Mlp::fit.
FitReport
Output of a training run.

Enums§

LrSchedule
Learning rate schedule (applied per epoch).
Shuffle
Training data shuffling strategy.