ray/release/rllib_tests/learning_tests/yaml_files/dqn/dqn-breakoutnoframeskip-v4.yaml
Avnish Narayan 82395c4646
[RLlib] Put learning test into own folders (#26862)
Co-authored-by: Artur Niederfahrenhorst <artur@anyscale.com>
2022-07-22 11:20:47 -07:00

31 lines
909 B
YAML

dqn-breakoutnoframeskip-v4:
env: BreakoutNoFrameskip-v4
run: DQN
# Minimum reward and total ts (in given time_total_s) to pass this test.
pass_criteria:
episode_reward_mean: 20.0
timesteps_total: 400000
stop:
time_total_s: 7200
config:
double_q: false
dueling: false
num_atoms: 1
noisy: false
n_step: 1
target_network_update_freq: 8000
lr: .0000625
adam_epsilon: .00015
hiddens: [512]
replay_buffer_config:
type: MultiAgentReplayBuffer
capacity: 1000000
learning_starts: 20000
prioritized_replay_alpha: 0.5
rollout_fragment_length: 4
train_batch_size: 32
exploration_config:
epsilon_timesteps: 200000
final_epsilon: 0.01
num_gpus: 0.5
min_sample_timesteps_per_iteration: 10000