revert benchmark code to include all agents/envs

cpnota · Mar 6, 2024 · 83ce920 · 83ce920
1 parent 67d6904
commit 83ce920
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 10 deletions.
diff --git a/all/environments/pybullet.py b/all/environments/pybullet.py
@@ -5,8 +5,8 @@ class PybulletEnvironment(GymEnvironment):
     short_names = {
         "ant": "AntBulletEnv-v0",
         "cheetah": "HalfCheetahBulletEnv-v0",
-        "humanoid": "HumanoidBulletEnv-v0",
         "hopper": "HopperBulletEnv-v0",
+        "humanoid": "HumanoidBulletEnv-v0",
         "walker": "Walker2DBulletEnv-v0",
     }
 

diff --git a/benchmarks/mujoco_v4.py b/benchmarks/mujoco_v4.py
@@ -8,16 +8,14 @@ def main():
 
     agents = [ddpg, ppo, sac]
 
-    agents = [sac]
-
     envs = [
         MujocoEnvironment(env, device="cuda")
         for env in [
-            # "Ant-v4",
+            "Ant-v4",
             "HalfCheetah-v4",
-            # "Hopper-v4",
-            # "Humanoid-v4",
-            # "Walker2d-v4",
+            "Hopper-v4",
+            "Humanoid-v4",
+            "Walker2d-v4",
         ]
     ]
 
@@ -27,7 +25,7 @@ def main():
         frames,
         logdir="benchmarks/mujoco_v4",
         sbatch_args={
-            "partition": "gypsum-2080ti",
+            "partition": "gpu-long",
         },
     )
 

diff --git a/benchmarks/pybullet.py b/benchmarks/pybullet.py
@@ -10,15 +10,23 @@ def main():
 
     envs = [
         PybulletEnvironment(env, device="cuda")
-        for env in PybulletEnvironment.short_names
+        for env in [
+            "AntBulletEnv-v0",
+            "HalfCheetahBulletEnv-v0",
+            "HopperBulletEnv-v0",
+            "HumanoidBulletEnv-v0",
+            "Walker2DBulletEnv-v0",
+        ]
     ]
 
     SlurmExperiment(
         agents,
         envs,
         frames,
         logdir="benchmarks/pybullet",
-        sbatch_args={"partition": "gpu-long"},
+        sbatch_args={
+            "partition": "gpu-long",
+        },
     )