ray.rllib.core.learner.learner.Learner.register_metric#
- Learner.register_metric(module_id: str, key: str, value: Any) None [source]#
Registers a single key/value metric pair for loss- and gradient stats.
- Parameters:
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).