Trait TrainingCallback 

pub trait TrainingCallback: Send + Sync {
    // Provided methods
    fn on_stage_change(&self, _stage: &PipelineStage) { ... }
    fn on_batch_complete(
        &self,
        _batch_idx: usize,
        _total_batches: usize,
        _avg_quality: f32,
    ) { ... }
    fn on_epoch_complete(&self, _epoch: usize, _stats: &EpochStats) { ... }
    fn on_training_complete(&self, _result: &TrainingResult) { ... }
    fn on_error(&self, _error: &str) { ... }
}

Callback trait for training events

Provided Methods

fn on_stage_change(&self, _stage: &PipelineStage)

Called when stage changes

fn on_batch_complete( &self, _batch_idx: usize, _total_batches: usize, _avg_quality: f32, )

Called after each batch

fn on_epoch_complete(&self, _epoch: usize, _stats: &EpochStats)

Called after each epoch

fn on_training_complete(&self, _result: &TrainingResult)

Called when training completes

fn on_error(&self, _error: &str)

Called on error

Implementors