Skip to main content

EventProcessorTraining

burn_train

Trait EventProcessorTraining 

Source 
pub trait EventProcessorTraining<TrainEvent, ValidEvent>: Send {
    // Required methods
    fn process_train(&mut self, event: TrainEvent);
    fn process_valid(&mut self, event: ValidEvent);
    fn renderer(self) -> Box<dyn MetricsRenderer>;
}
Expand description

Process events happening during training and validation.

Required Methods§

Source

fn process_train(&mut self, event: TrainEvent)

Collect a training event.

Source

fn process_valid(&mut self, event: ValidEvent)

Collect a validation event.

Source

fn renderer(self) -> Box<dyn MetricsRenderer>

Returns the renderer used for training.

Implementors§