diff --git a/experience_replay.py b/experience_replay.py index 923cd97..d26b98d 100644 --- a/experience_replay.py +++ b/experience_replay.py @@ -100,7 +100,7 @@ def sample(self, batch_size): self.rewards[idxs], self.next_states[idxs], self.dones[idxs], - self.priority[idxs]/sum(self.priority[idxs]), + self.priority[idxs]/sum(self.priority), idxs )