Trait TrainingCallback 

pub trait TrainingCallback {
    // Required methods
    fn on_epoch_start(&mut self, epoch: usize, numepochs: usize);
    fn on_epoch_end(
        &mut self,
        epoch: usize,
        numepochs: usize,
        metrics: &Metrics,
    );
    fn on_batch_start(&mut self, batch: usize, numbatches: usize);
    fn on_batch_end(&mut self, batch: usize, numbatches: usize, loss: f64);
    fn on_train_start(&mut self, numepochs: usize);
    fn on_train_end(&mut self, metrics: &Metrics);
}

Training progress callback trait.

Required Methods

fn on_epoch_start(&mut self, epoch: usize, numepochs: usize)

Called at the start of each epoch.

fn on_epoch_end(&mut self, epoch: usize, numepochs: usize, metrics: &Metrics)

Called at the end of each epoch.

fn on_batch_start(&mut self, batch: usize, numbatches: usize)

Called at the start of each batch.

fn on_batch_end(&mut self, batch: usize, numbatches: usize, loss: f64)

Called at the end of each batch.

fn on_train_start(&mut self, numepochs: usize)

Called at the start of training.

fn on_train_end(&mut self, metrics: &Metrics)

Called at the end of training.

Implementors

impl TrainingCallback for ParameterSyncCallback

impl TrainingCallback for ProgressCallback