composer.callbacks

Modules

composer.callbacks.callback_hparams_registry	Hyperparameter registry for callbacks.
composer.callbacks.checkpoint_saver	Callback to save checkpoints during training.
composer.callbacks.early_stopper	Early stopping callback.
composer.callbacks.grad_monitor	Monitor gradients during training.
composer.callbacks.image_visualizer	Monitor train and eval images.
composer.callbacks.lr_monitor	Monitor learning rate during training.
composer.callbacks.memory_monitor	Log memory usage during training.
composer.callbacks.mlperf	Create a submission for MLPerf Training benchmark.
composer.callbacks.speed_monitor	Monitor throughput during training.
composer.callbacks.threshold_stopper	Threshold stopping callback.

Callbacks that run at each training loop Event.

Each callback inherits from the Callback base class. See detailed description and examples for writing your own callbacks at the Callback base class.

Classes

CheckpointSaver	Callback to save checkpoints.
EarlyStopper	Track a metric and halt training if it does not improve within a given interval.
GradMonitor	Computes and logs the L2 norm of gradients on the Event.AFTER_TRAIN_BATCH event.
ImageVisualizer	Logs image inputs and optionally outputs.
LRMonitor	Logs the learning rate.
MLPerfCallback	Create compliant results file for MLPerf Training benchmark.
MemoryMonitor	Logs the memory usage of the model.
SpeedMonitor	Logs the training throughput.
ThresholdStopper	Halt training when a metric value reaches a certain threshold.