baselines
baselines copied to clipboard
Added custom logging to PPO2
In our work we need to log custom information. While a monitored environment already allows us to record this information, we would like to be able to track the data on tensorboard along with the rest of the data PPO2 writes out.
This PR allows you to specify a tuple of extra keys to be logged from the episode info.