composer.callbacks.callback_hparams_registry#

Hyperparameter registry for callbacks.

Classes

Callback

Base class for callbacks.

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.

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.

Attributes

  • Dict

  • Type

  • Union

  • callback_registry