Encoder decoder loss plot

danijar / dreamerv3

Mastering Diverse Domains through World Models

https://danijar.com/dreamerv3

MIT License

1.28k stars 219 forks source link

Open kubs0ne opened 1 year ago

kubs0ne commented 1 year ago

Hello, is it possible to visualise the loss plot of the encoder-decoder? I am new to jax so not sure how to do it. Regards