[Question] How to render "info" on tensorboard

[SKYLEO98]

❓ Question

Hello,

The "info" return from my custom env will cover the partial rewards. I would like to render them on tensor board. Is there any way to update the output of the tensor board log?
Many thanks in advance.

Checklist

• I have checked that there is no similar issue in the repo
• I have read the SB3 documentation
• I have read the RL Zoo documentation
• If code there is, it is minimal and working
• If code there is, it is formatted using the markdown code blocks for both code and stack traces.

[araffin]

Hello,
you can use callback for that:

rl-baselines3-zoo/rl_zoo3/callbacks.py

Line 207 in 27e081e

class RawStatisticsCallback(BaseCallback):

See documentation too.