add deterministic Atari v3 benchmark

angrySquirrel · Nov 22, 2016 · 3bee100 · 3bee100
1 parent 95f1248
commit 3bee100
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/gym/benchmarks/__init__.py b/gym/benchmarks/__init__.py
@@ -75,6 +75,42 @@
         }
     ])
 
+register_benchmark(
+    id='Atari7PixelDeterministic-v3',
+    scorer=scoring.ClipTo01ThenAverage(),
+    name='Atari7PixelDeterministic-v3',
+    description='7 Atari games, with pixel observations',
+    tasks=[
+        {'env_id': 'BeamRiderDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        },
+        {'env_id': 'BreakoutDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        },
+        {'env_id': 'EnduroDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        },
+        {'env_id': 'PongDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        },
+        {'env_id': 'QbertDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        },
+        {'env_id': 'SeaquestDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        },
+        {'env_id': 'SpaceInvadersDeterministic-v3',
+         'trials': 2,
+         'max_timesteps': 25000000,
+        }
+    ])
+
 register_benchmark(
     id='Atari7Pixel-v3',
     scorer=scoring.ClipTo01ThenAverage(),