Rewards by curiosity module

RLlib
1

Hi I am using the curiosity module with the PPO algorithm and wanted to know if there is a way I can log the rewards generated by the curiosity module only rather than the total reward?

It would be a great help to understand the effect of curiosity module.

2

@naren_dhyani , there is. You could write a custom metrics callback that takes care of this. See for a use case ray/rllib/examples/custom_metrics_and_callbacks.py.