dm_hamiltonian_dynamics_suite
dm_hamiltonian_dynamics_suite copied to clipboard
Need dataset keys clarification
I see that reach datapoint has the following keys - dict_keys(['dx_dt', 'image', 'other', 'x']).
In the implementation of Physics inspired models, only image key has been used.
I want to use this dataset, but not as images. My questions are:
- Is it possible to use trajectories of all the 17 datasets without images?
- Do the keys
x, dx_dtrefer to trajectory points and derivatives?
We use images, as our focus in this was only on models that learn from images. Answering your questions:
- Yes, there is no issue with using them without the images. The images were added so that there is consistent rendering if people use them.
- For the toy problem datasets
xis the concatenation of position and momenta, anddx_dtis their derivative w.r.t time, for the multi-agent datasets I think the state is the population policies, for mujoco rooms the state is the position of the camera and for molecular dynamics the state is the particle positions and momenta as well.