Struct ParameterSyncCallback 

pub struct ParameterSyncCallback { /* private fields */ }

Callback for synchronizing parameters between workers.

Implementations

impl ParameterSyncCallback

pub fn new( syncinterval: usize, channel: Box<dyn DistributedCommunication>, ) -> Self

Create a new parameter synchronization callback.

Trait Implementations

impl TrainingCallback for ParameterSyncCallback

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, _num_epochs: 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.