cartpole-dqn-fake-gpus.yaml 444 B

12345678910111213141516171819
  1. cartpole-dqn-fake-gpus:
  2. env: CartPole-v0
  3. run: DQN
  4. stop:
  5. episode_reward_mean: 150
  6. training_iteration: 400
  7. config:
  8. # Works for both torch and tf.
  9. framework: tf
  10. model:
  11. fcnet_hiddens: [64]
  12. fcnet_activation: linear
  13. n_step: 3
  14. # Double batch size (2 GPUs).
  15. train_batch_size: 64
  16. # Fake 2 GPUs.
  17. num_gpus: 2
  18. _fake_gpus: true