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§
Sourcefn on_epoch_end(&self, epoch: usize, history: &TrainingHistory) -> Result<()>
fn on_epoch_end(&self, epoch: usize, history: &TrainingHistory) -> Result<()>
Called at the end of each epoch