Skip to content

Commit 430f634

Browse files
committed
Updated hyper params
1 parent 0c65d4e commit 430f634

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

intermediate_source/reinforcement_q_learning.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -266,11 +266,12 @@ def forward(self, x):
266266

267267
BATCH_SIZE = 128
268268
GAMMA = 0.99
269-
EPS_START = 1
269+
EPS_START = 0.9
270270
EPS_END = 0.01
271271
EPS_DECAY = 2500
272272
TAU = 0.005
273-
LR = 5e-4
273+
LR = 3e-4
274+
274275

275276
# Get number of actions from gym action space
276277
n_actions = env.action_space.n

0 commit comments

Comments
 (0)