Create dqn.yml
Browse files
dqn.yml
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
SpaceInvadersNoFrameskip-v4:
|
2 |
+
env_wrapper:
|
3 |
+
- stable_baselines3.common.atari_wrappers.AtariWrapper
|
4 |
+
frame_stack: 4
|
5 |
+
policy: 'CnnPolicy'
|
6 |
+
n_timesteps: !!float 1e4
|
7 |
+
buffer_size: 100000
|
8 |
+
learning_rate: !!float 1e-4
|
9 |
+
batch_size: 32
|
10 |
+
learning_starts: 100000
|
11 |
+
target_update_interval: 1000
|
12 |
+
train_freq: 4
|
13 |
+
gradient_steps: 1
|
14 |
+
exploration_fraction: 0.1
|
15 |
+
exploration_final_eps: 0.01
|
16 |
+
# If True, you need to deactivate handle_timeout_termination
|
17 |
+
# in the replay_buffer_kwargs
|
18 |
+
optimize_memory_usage: False
|