Is there a way to get the .xml file of the locomotion soccer environment? Run environment in GPU?

google-deepmind / dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

Apache License 2.0

3.82k stars 669 forks source link

Is there a way to get the .xml file of the locomotion soccer environment? Run environment in GPU? #486

Open StefanoBraghetto opened 3 months ago

StefanoBraghetto commented 3 months ago

I want to create a project from this environment: soccer But I can't get with the .xml. My final objective is to run it in the GPU so I can run multiple instance of it at the same time. I was reading that this is possible with mujoco_mjx and I could load the environment doing:

mj_model = mujoco.MjModel.from_xml_path(
        (path / 'name_file.xml').as_posix())

please help!

sibocw commented 1 month ago

I think #429 gives the answer, unless the developers have since decided to retrofit MJX?

kevinzakka commented 1 month ago

@sibocw @StefanoBraghetto I have an implementation of dm_control suite that is fully based on MJX and will be open-sourced very soon.

sibocw commented 4 weeks ago

@kevinzakka Awesome! Looking forward :)