Hi, I’ve used callbacks before to measure custom metrics on episodes. The way I usually do it is by implementing the on_episode_end method, and adding a line like this: (env,) = base_env.get_sub_environments() episode.custom_metrics |= { 'my_metric': env.my_metric, } This works, and then lat…

Accessing custom metrics for episodes

cool-RR January 21, 2023, 9:02pm 2

Oh, I just found keep_per_episode_custom_metrics=True, so this solves my problem.

1 Like

How to obtain single episode reward?

Topic		Replies	Views
Custom metrics only mean value RLlib	3	787	February 16, 2022
Logging custom arrays with RLlib+Tune RLlib	2	704	June 16, 2021
Is this the recommended way of saving per-episode training metrics? RLlib	1	341	February 6, 2024
Custom_metrics not reporting as expected RLlib	4	284	March 3, 2021
MAML - Custom Metrics not reported RLlib	0	246	November 19, 2021