atari-dist-dqn.yaml 899 B

12345678910111213141516171819202122232425262728293031
  1. atari-dist-dqn:
  2. env:
  3. grid_search:
  4. - BreakoutNoFrameskip-v4
  5. - BeamRiderNoFrameskip-v4
  6. - QbertNoFrameskip-v4
  7. - SpaceInvadersNoFrameskip-v4
  8. run: DQN
  9. config:
  10. double_q: false
  11. dueling: false
  12. num_atoms: 51
  13. noisy: false
  14. prioritized_replay: false
  15. n_step: 1
  16. target_network_update_freq: 8000
  17. lr: .0000625
  18. adam_epsilon: .00015
  19. hiddens: [512]
  20. learning_starts: 20000
  21. buffer_size: 1000000
  22. rollout_fragment_length: 4
  23. train_batch_size: 32
  24. exploration_config:
  25. epsilon_timesteps: 200000
  26. final_epsilon: 0.01
  27. prioritized_replay_alpha: 0.5
  28. final_prioritized_replay_beta: 1.0
  29. prioritized_replay_beta_annealing_timesteps: 2000000
  30. num_gpus: 0.2
  31. timesteps_per_iteration: 10000