Accessing mean reward for curriculum learning

[MarcDcls]

Hi,

I’m currently exploring curriculum learning strategies with mjlab, and I was wondering whether there is an existing way to access the mean reward during training.

At the moment, I’m basing the curriculum solely on the number of training steps, which works to some extent but is not very satisfying compared to a performance-based signal.

Thanks in advance!

[kevinzakka]

Hi @MarcDcls,

The answer is yes! Curriculum functions receive env as their first argument, so you have access to all managers, including the reward manager. You can compute the mean reward with something like:

env.reward_manager._episode_sums["your_reward_term"][env_ids].mean()

Cheers

[MarcDcls]

Nice 👍

Thanks for the quick answer!