ray/rllib/tuned_examples/dqn/cartpole-dqn.yaml

11 lines
233 B
YAML

cartpole-dqn:
env: CartPole-v0
run: DQN
stop:
episode_reward_mean: 150
timesteps_total: 50000
config:
# Works for both torch and tf.
framework: tf
n_step: 3
gamma: 0.95