callback_hparams_registry#
Hyperparameter registry for callbacks.
Classes
Base class for callbacks. |
|
Callback to save checkpoints. |
|
Track a metric and halt training if it does not improve within a given interval. |
|
Callback to export model for inference. |
|
Computes and logs the L2 norm of gradients on the |
|
Logs image inputs and optionally outputs. |
|
Logs the learning rate. |
|
Create compliant results file for MLPerf Training benchmark. |
|
Logs the memory usage of the model. |
|
Logs the training throughput. |
|
Halt training when a metric value reaches a certain threshold. |
Attributes
Dict
Type
Union
callback_registry