How can I get the episode info dict from the on_sample_end callback?

138 Views Asked by At

I need to get the episode info dict from the on_sample_end callback so that I can display some custom metrics every time a rollout has finished. How can I do that? Ray Version: 0.7.3

Thank you in advance!

0

There are 0 best solutions below