composer.callbacks

Modules

composer.callbacks.callback_hparams	Hyperparameters for callbacks.
composer.callbacks.grad_monitor	Monitor gradient during training.
composer.callbacks.lr_monitor	Monitor learning rate during training.
composer.callbacks.memory_monitor	Log memory usage during training.
composer.callbacks.run_directory_uploader	Periodically upload run_directory to a blob store during training.
composer.callbacks.speed_monitor	Monitor throughput during training.

Callbacks provide hooks that can 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

GradMonitor	Computes and logs the L2 norm of gradients on the AFTER_TRAIN_BATCH event.
LRMonitor	Logs the learning rate.
MemoryMonitor	Logs the memory usage of the model.
RunDirectoryUploader	Callback to upload the run directory to a blob store.
SpeedMonitor	Logs the training throughput.