Callbacks for RL agents

[team-daniel]

Hi,

I was wondering if callbacks can be made like they can in StableBaselines3 so I can monitor custom metrics during training?

Thanks,

[liuzuxin]

What customized metrics are you referring to? During data collection or policy updating?

[team-daniel]

Thanks for the reply! Yes I mean during data collection (at the end of rollouts for example)

[liuzuxin]

Do you think this preprocess fn here helpful?

If not, the policy_udpate_fn here and the gather_update_info function here could be overrides to collect the episode-wise stats.