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.
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.