ray/rllib/tuned_examples/simple_q/cartpole-simpleq-fake-gpus.yaml

13 lines
283 B
YAML

cartpole-simpleq-fake-gpus:
env: CartPole-v0
run: SimpleQ
stop:
episode_reward_mean: 150
training_iteration: 400
config:
# Works for both torch and tf.
framework: tf
# Fake 2 GPUs.
num_gpus: 2
_fake_gpus: true