[rllib] Propagate dim option to deepmind wrappers (#1876)

* updates

* updates
This commit is contained in:
Eric Liang 2018-04-11 21:38:06 -07:00 committed by GitHub
parent 15a668dd12
commit 4dc04374f6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions

View file

@ -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)

View file

@ -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