Struct LearnerCheckpointer 

pub struct LearnerCheckpointer<LC: LearnerComponentTypes> { /* private fields */ }

Used to create, delete, or load checkpoints of the training process.

Implementations

impl<LC: LearnerComponentTypes> LearnerCheckpointer<LC>

pub fn new( model: LC::CheckpointerModel, optim: LC::CheckpointerOptimizer, lr_scheduler: LC::CheckpointerLrScheduler, strategy: LC::CheckpointerStrategy, ) -> Self

Constructs a new LearnerCheckpointer.

impl<LC: LearnerComponentTypes> LearnerCheckpointer<LC>

pub fn checkpoint( &mut self, model: &LC::Model, optim: &LC::Optimizer, scheduler: &LC::LrScheduler, epoch: usize, store: &EventStoreClient, )

Create checkpoint for the training process.

pub fn load_checkpoint( &self, model: LC::Model, optim: LC::Optimizer, scheduler: LC::LrScheduler, device: &Device<LC::Backend>, epoch: usize, ) -> (LC::Model, LC::Optimizer, LC::LrScheduler)

Load a training checkpoint.