Trait burn::train::renderer::MetricsRenderer

pub trait MetricsRenderer: Send + Sync {
    // Required methods
    fn update_train(&mut self, state: MetricState);
    fn update_valid(&mut self, state: MetricState);
    fn render_train(&mut self, item: TrainingProgress);
    fn render_valid(&mut self, item: TrainingProgress);
}
Expand description

Trait for rendering metrics.

Required Methods§

fn update_train(&mut self, state: MetricState)

Updates the training metric state.

§Arguments
  • state - The metric state.

fn update_valid(&mut self, state: MetricState)

Updates the validation metric state.

§Arguments
  • state - The metric state.

fn render_train(&mut self, item: TrainingProgress)

Renders the training progress.

§Arguments
  • item - The training progress.

fn render_valid(&mut self, item: TrainingProgress)

Renders the validation progress.

§Arguments
  • item - The validation progress.

Implementors§