mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-22 02:25:12 -05:00
19 lines
488 B
YAML
19 lines
488 B
YAML
pong-impala-vectorized:
|
|
env: ALE/Pong-v5
|
|
run: IMPALA
|
|
config:
|
|
# Make analogous to old v4 + NoFrameskip.
|
|
env_config:
|
|
frameskip: 1
|
|
full_action_space: false
|
|
repeat_action_probability: 0.0
|
|
rollout_fragment_length: 50
|
|
train_batch_size: 500
|
|
num_workers: 11
|
|
num_envs_per_worker: 10
|
|
framework: torch
|
|
log_level: INFO
|
|
stop:
|
|
episode_reward_mean: 10
|
|
time_total_s: 3600
|