Trait burn::train::EarlyStoppingStrategy

pub trait EarlyStoppingStrategy {
    // Required method
    fn should_stop(&mut self, epoch: usize, store: &EventStoreClient) -> bool;
}

A strategy that checks if the training should be stopped.

Required Methods

fn should_stop(&mut self, epoch: usize, store: &EventStoreClient) -> bool

Update its current state and returns if the training should be stopped.

Implementors

impl EarlyStoppingStrategy for MetricEarlyStoppingStrategy