ray/rllib/tuned_examples/a3c/cartpole-a2c-fake-gpus.yaml

14 lines
315 B
YAML

cartpole-a2c-fake-gpus:
env: CartPole-v0
run: A2C
stop:
episode_reward_mean: 150
training_iteration: 200
config:
# Works for both torch and tf.
framework: tf
num_workers: 0
lr: 0.001
# Fake 2 GPUs.
num_gpus: 2
_fake_gpus: true