mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
[rllib] Propagate dim option to deepmind wrappers (#1876)
* updates * updates
This commit is contained in:
parent
15a668dd12
commit
4dc04374f6
2 changed files with 2 additions and 4 deletions
|
@ -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)
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Add table
Reference in a new issue