From c6a0ae6b4c593fd8c828d3a47fd282ff4c4ce56d Mon Sep 17 00:00:00 2001 From: Wenpu Wang Date: Wed, 20 Apr 2022 02:43:02 -0400 Subject: [PATCH] fixed bugs modify update_priorities to avoid zip error --- dqn.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/dqn.py b/dqn.py index 9cf9336..01108b3 100644 --- a/dqn.py +++ b/dqn.py @@ -87,8 +87,10 @@ def sample(self, frame_idx, batch_size=20, beta_frames=3000): action_samples, reward_samples, next_state_samples, done_samples, indices, weights def update_priorities(self, batch_indices, batch_priorities): - for idx, prio in zip(batch_indices, batch_priorities): - self.priorities[idx] = prio + # for idx, prio in zip(batch_indices, batch_priorities): + # self.priorities[idx] = prio + for idx in batch_indices: + self.priorities[idx] = batch_priorities def __len__(self): return self.cnt