Skip to main content

Module training

Module training 

Source
Available on crate feature std only.
Expand description

Combined model, optimizer, scheduler and accumulated-gradient records.

Structsยง

RestoredTraining
Components restored together, including pending gradients and caller state.
TrainingRecord
One record containing the trainable state and a caller-defined continuation record.