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

26 lines
736 B
YAML
Raw Normal View History

stateless-cartpole-r2d2:
env: ray.rllib.examples.env.stateless_cartpole.StatelessCartPole
2021-02-25 12:18:11 +01:00
run: R2D2
stop:
episode_reward_mean: 150
timesteps_total: 1000000
config:
# Works for both torch and tf.
framework: tf
num_workers: 0
# R2D2 settings.
burn_in: 20
zero_init_states: true
#dueling: false
2021-02-25 12:18:11 +01:00
lr: 0.0005
# Give some more time to explore.
exploration_config:
epsilon_timesteps: 50000
2021-02-25 12:18:11 +01:00
# Wrap with an LSTM and use a very simple base-model.
model:
fcnet_hiddens: [64]
fcnet_activation: linear
2021-02-25 12:18:11 +01:00
use_lstm: true
lstm_cell_size: 64
max_seq_len: 20