From 4dc04374f6fa9646f84cee87661c4107b200a52f Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Wed, 11 Apr 2018 21:38:06 -0700 Subject: [PATCH] [rllib] Propagate dim option to deepmind wrappers (#1876) * updates * updates --- python/ray/rllib/dqn/common/wrappers.py | 3 ++- python/ray/rllib/tuned_examples/pong-dqn.yaml | 3 --- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/python/ray/rllib/dqn/common/wrappers.py b/python/ray/rllib/dqn/common/wrappers.py index 9a2d0d764..a968888aa 100644 --- a/python/ray/rllib/dqn/common/wrappers.py +++ b/python/ray/rllib/dqn/common/wrappers.py @@ -14,6 +14,7 @@ def wrap_dqn(registry, env, options, random_starts): # Override atari default to use the deepmind wrappers. # TODO(ekl) this logic should be pushed to the catalog. if is_atari and "custom_preprocessor" not in options: - return wrap_deepmind(env, random_starts=random_starts) + return wrap_deepmind( + env, random_starts=random_starts, dim=options.get("dim", 80)) return ModelCatalog.get_preprocessor_as_wrapper(registry, env, options) diff --git a/python/ray/rllib/tuned_examples/pong-dqn.yaml b/python/ray/rllib/tuned_examples/pong-dqn.yaml index 51e1bd2e6..c22020c67 100644 --- a/python/ray/rllib/tuned_examples/pong-dqn.yaml +++ b/python/ray/rllib/tuned_examples/pong-dqn.yaml @@ -2,9 +2,6 @@ pong-deterministic-dqn: env: PongDeterministic-v4 run: DQN - trial_resources: - cpu: 1 - gpu: 1 stop: episode_reward_mean: 20 time_total_s: 7200