Trait burn::train::checkpoint::CheckpointingStrategy

pub trait CheckpointingStrategy {
    // Required method
    fn checkpointing(
        &mut self,
        epoch: usize,
        collector: &EventStoreClient,
    ) -> Vec<CheckpointingAction>;
}

Define when checkpoint should be saved and deleted.

Required Methods

fn checkpointing( &mut self, epoch: usize, collector: &EventStoreClient, ) -> Vec<CheckpointingAction>

Based on the epoch, determine if the checkpoint should be saved.

Trait Implementations

impl CheckpointingStrategy for Box<dyn CheckpointingStrategy>

fn checkpointing( &mut self, epoch: usize, collector: &EventStoreClient, ) -> Vec<CheckpointingAction>

Based on the epoch, determine if the checkpoint should be saved.

Implementations on Foreign Types

impl CheckpointingStrategy for Box<dyn CheckpointingStrategy>

fn checkpointing( &mut self, epoch: usize, collector: &EventStoreClient, ) -> Vec<CheckpointingAction>

Implementors

impl CheckpointingStrategy for ComposedCheckpointingStrategy

impl CheckpointingStrategy for KeepLastNCheckpoints

impl CheckpointingStrategy for MetricCheckpointingStrategy