Callback

Trait Callback 

Source
pub trait Callback: Send + Sync {
    // Required method
    fn on_epoch_end(
        &self,
        epoch: usize,
        history: &TrainingHistory,
    ) -> Result<()>;
}
Expand description

Callback trait for training

Required Methods§

Source

fn on_epoch_end(&self, epoch: usize, history: &TrainingHistory) -> Result<()>

Called at the end of each epoch

Implementors§