mirror of
https://github.com/wassname/ray.git
synced 2026-06-30 09:41:11 +08:00
[rllib] Propagate dim option to deepmind wrappers (#1876)
* updates * updates
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user