andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Morvan Zhou
2017-05-29 16:04:38 +10:00
committed by Morvan Zhou
parent b2922fd3f3
commit cc831bf4ef
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tutorial-contents/405_DQN_Reinforcement_learning.py
View File

@ -74,6 +74,7 @@ class DQN(object):
# target parameter update # target parameter update
if self.learn_step_counter % TARGET_REPLACE_ITER == 0: if self.learn_step_counter % TARGET_REPLACE_ITER == 0:
self.target_net.load_state_dict(self.eval_net.state_dict()) self.target_net.load_state_dict(self.eval_net.state_dict())
self.learn_step_counter += 1
# sample batch transitions # sample batch transitions
sample_index = np.random.choice(MEMORY_CAPACITY, BATCH_SIZE) sample_index = np.random.choice(MEMORY_CAPACITY, BATCH_SIZE)