title: Stable-Baselines3 log rewards
body: <p>How can I add the rewards to tensorboard logging in Stable Baselines3 using a custom environment?</p>
  <p>I have this learning code</p>
  <pre><code>model = PPO(
      &quot;MlpPolicy&quot;, env,
      learning_rate=1e-4,
      policy_kwargs=policy_kwargs,
      verbose=1,
      tensorboard_log=&quot;./tensorboard/&quot;)
  </code></pre>