Add dm-control

This commit is contained in:
Simon Alibert
2024-03-14 13:42:03 +01:00
parent c3f8d14fd8
commit f515cb6efd
4 changed files with 501 additions and 177 deletions

View File

@@ -42,13 +42,14 @@ mpmath = "^1.3.0"
torch = "^2.2.1"
tensordict = {git = "https://github.com/pytorch/tensordict"}
torchrl = {git = "https://github.com/pytorch/rl", rev = "13bef426dcfa5887c6e5034a6e9697993fa92c37"}
mujoco = "^3.1.2"
mujoco = "2.3.7"
mujoco-py = "^2.1.2.14"
gym = "^0.26.2"
opencv-python = "^4.9.0.80"
diffusers = "^0.26.3"
torchvision = "^0.17.1"
h5py = "^3.10.0"
dm-control = "1.0.14"
[tool.poetry.group.dev.dependencies]