ray.rllib.models.modelv2.ModelV2.metrics#

ModelV2.metrics() → Dict[str, numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor][source]#

Override to return custom metrics from your model.

The stats will be reported as part of the learner stats, i.e., info.learner.[policy_id, e.g. “default_policy”].model.key1=metric1