Learner.register_metric(module_id: str, key: str, value: Any) None[source]#

Registers a single key/value metric pair for loss- and gradient stats.

  • module_id – The module_id to register the metric under. This may be ALL_MODULES.

  • key – The name of the metric to register (below the given module_id).

  • value – The actual value of the metric. This might also be a tensor var (e.g. from within a traced tf2 function).