Fix typo in agent.py (#27)

xingdi-eric-yuan · Nov 14, 2021 · c1afc3c · c1afc3c
1 parent 1dc673d
commit c1afc3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent.py b/agent.py
@@ -1419,7 +1419,7 @@ def get_drqn_loss(self, episode_no):
             prev_h, prev_c = new_h, new_c
             if step_no < self.replay_sample_update_from:
                 q_value = q_value.detach()
-                prev_h, prev_c = prev_h.detach(), prev_h.detach()
+                prev_h, prev_c = prev_h.detach(), prev_c.detach()
                 continue
 
             with torch.no_grad():