Trait TrainCallback 

pub trait TrainCallback {
    // Required method
    fn on_epoch(
        &mut self,
        epoch: usize,
        total: usize,
        loss: f32,
        val_loss: Option<f32>,
    ) -> bool;
}

Required Methods

fn on_epoch( &mut self, epoch: usize, total: usize, loss: f32, val_loss: Option<f32>, ) -> bool

Implementors

impl TrainCallback for PrintCallback

impl<F> TrainCallback for F

where F: FnMut(usize, usize, f32, Option<f32>) -> bool,