facebookresearch / mbrl-lib

Library for Model Based RL
MIT License
959 stars 158 forks source link

Plotting #47

Closed natolambert closed 3 years ago

natolambert commented 3 years ago

Added notebook and function to plot reward over trials (trials can be replaced by anything, such as training epoch with a callback to log data).

Also, I would like to eventually include more MBRL diagnostics / plots in similar notebooks. May want to have another area for it, not sure everyone wants all the plots to be generated in Jupiter and it can be weird to run on remote.

Can you post how you run jupyter from remote to local? I forgot and just did this locally.