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.

ExportForInferenceCallback

Callback to export model for inference.

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.

Attributes

  • Dict

  • Type

  • Union

  • callback_registry