Struct ProgressLogger 

pub struct ProgressLogger { /* private fields */ }

Simple progress logging callback.

Implementations

impl ProgressLogger

pub fn new(log_every: usize) -> Self

Creates a new progress logger.

Trait Implementations

impl Callback for ProgressLogger

fn on_epoch_begin(&mut self, epoch: usize, _state: &TrainingState)

Called at the start of an epoch.

fn on_step_end( &mut self, step: usize, metrics: &TrainingMetrics, _state: &TrainingState, )

Called after each training step.

fn on_epoch_end(&mut self, epoch: usize, state: &TrainingState) -> bool

Called at the end of an epoch.

fn on_train_begin(&mut self, _state: &TrainingState)

Called at the start of training.

fn on_train_end(&mut self, _state: &TrainingState)

Called at the end of training.

fn on_validation_end( &mut self, _metrics: &TrainingMetrics, _state: &TrainingState, )

Called after validation.