google-research / planet

Learning Latent Dynamics for Planning from Pixels
https://danijar.com/planet
Apache License 2.0
1.18k stars 202 forks source link

Prediction vs Ground Truth #44

Closed pulver22 closed 5 years ago

pulver22 commented 5 years ago

Hi, after your latest push I noticed that the summary from comparing the planet prediction against the Mujoco ground truth have been removed. Is it correct? Can you explain what you did it and if is there any way to keep them without reverting to the previous commit?

danijar commented 5 years ago

Thanks for bringing this to my attention, and sorry for the slow reply. They are disabled to have the training and summary computation run a bit faster. However, it should be easy to enable them again. You can enable {state_diagnostics: True}. However, the only manual change right now is that you'll also need to create optimizers in configs.py for each state component. I'll set up a default optimizer that is used for them in the next update.