Struct RestoredTraining 

pub struct RestoredTraining<M, O, S, U> {
    pub model: M,
    pub optimizer: O,
    pub scheduler: S,
    pub accumulator: GradientsAccumulator<M>,
    pub state: U,
}

Available on crate feature std only.

Components restored together, including pending gradients and caller state.

Fields

model: M

Model with the recorded parameter IDs.

optimizer: O

Optimizer with the recorded parameter state.

scheduler: S

Scheduler at the recorded position.

accumulator: GradientsAccumulator<M>

Pending gradients, without an implicit optimizer update or reset.

state: U

Caller-defined continuation state.