raisimTech / raisimLib

Visit www.raisim.com
http://www.raisim.com
Other
341 stars 91 forks source link

How to add seperate reward terms to tensorboard #494

Open P1terQ opened 1 year ago

P1terQ commented 1 year ago

In environment.hpp, i define 4 reward terms and i want to log them in tensorboard. However, in runner.py, it seems that reward, dones = env.step(action) only returns rewards.sum() in each environment. It's inconvenient to tune the reward coefficients . How can i get the seperate values of reward terms instead of their sum? Looking forward to your reply. Much thanks.

jhwangbo commented 1 year ago

I already do this in my research code. I'll add this feature in the next push.

P1terQ commented 1 year ago

Thanks a lot! looking forward to it!

jhwangbo commented 1 year ago

I'll keep it open so that I don't forget to reflect this

jhwangbo commented 1 year ago

I added the feature. Can you pull and test?