Trait CheckpointSupport 

pub trait CheckpointSupport {
    // Required methods
    fn save_state_for_checkpoint(&self) -> Result<Vec<u8>, OptimizerError>;
    fn load_state_from_checkpoint(
        &mut self,
        data: &[u8],
    ) -> Result<(), OptimizerError>;
    fn get_gradients_for_checkpoint(&self) -> Option<HashMap<String, Vec<f32>>>;
    fn get_parameters_for_checkpoint(&self) -> Option<HashMap<String, Vec<f32>>>;
}

Trait for optimizers that support checkpointing

Required Methods

fn save_state_for_checkpoint(&self) -> Result<Vec<u8>, OptimizerError>

Save optimizer state for checkpointing

fn load_state_from_checkpoint( &mut self, data: &[u8], ) -> Result<(), OptimizerError>

Load optimizer state from checkpoint

fn get_gradients_for_checkpoint(&self) -> Option<HashMap<String, Vec<f32>>>

Get current gradients for checkpointing

fn get_parameters_for_checkpoint(&self) -> Option<HashMap<String, Vec<f32>>>

Get current model parameters for checkpointing

Dyn Compatibility

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors