impala-breakoutnoframeskip-v5.yaml 663 B

12345678910111213141516171819202122
  1. impala-breakoutnoframeskip-v5:
  2. env: ALE/Breakout-v5
  3. run: IMPALA
  4. # Minimum reward and total ts (in given time_total_s) to pass this test.
  5. pass_criteria:
  6. sampler_results/episode_reward_mean: 200.0
  7. timesteps_total: 6000000
  8. stop:
  9. time_total_s: 2400
  10. config:
  11. # Make analogous to old v4 + NoFrameskip.
  12. env_config:
  13. frameskip: 1
  14. full_action_space: false
  15. repeat_action_probability: 0.0
  16. rollout_fragment_length: 50
  17. train_batch_size: 500
  18. num_workers: 10
  19. num_envs_per_worker: 5
  20. clip_rewards: True
  21. lr: 0.0005
  22. num_gpus: 1