medkit.training.callbacks#

Classes:

`DefaultPrinterCallback`()	Default implementation of `TrainerCallback`
`TrainerCallback`()	A TrainerCallback is the base class for trainer callbacks

class TrainerCallback[source]#

A TrainerCallback is the base class for trainer callbacks

Methods:

`on_epoch_begin`(epoch)	Event called at the beginning of an epoch
`on_epoch_end`(metrics, epoch, epoch_time)	Event called at the end of an epoch
`on_save`(checkpoint_dir)	Event called on saving a checkpoint
`on_step_begin`(step_idx, nb_batches, phase)	Event called at the beginning of a step in training
`on_step_end`(step_idx, nb_batches, phase)	Event called at the end of a step in training
`on_train_begin`(config)	Event called at the beginning of training
`on_train_end`()	Event called at the end of training

on_train_begin(config)[source]#

Event called at the beginning of training

on_epoch_begin(epoch)[source]#

Event called at the beginning of an epoch

on_epoch_end(metrics, epoch, epoch_time)[source]#

Event called at the end of an epoch

on_step_begin(step_idx, nb_batches, phase)[source]#

Event called at the beginning of a step in training

on_step_end(step_idx, nb_batches, phase)[source]#

Event called at the end of a step in training

on_save(checkpoint_dir)[source]#

Event called on saving a checkpoint

class DefaultPrinterCallback[source]#

Default implementation of TrainerCallback