Learner.register_metrics(module_id: str, metrics_dict: Dict[str, Any]) None[source]#

Registers several key/value metric pairs for loss- and gradient stats.

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

  • metrics_dict – A dict mapping names of metrics to be registered (below the given module_id) to the actual values of these metrics. Values might also be tensor vars (e.g. from within a traced tf2 function). These will be automatically converted to numpy values.