Update replay_buffer.py

Fixed bug

Update replay_buffer.py
Fixed bug
55f3f240 · Jaeyoun Kim · GitHub · 5f1c1b09 · 55f3f240
Unverified Commit 55f3f240 authored Apr 24, 2020 by Jaeyoun Kim Committed by GitHub Apr 24, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/pcl_rl/replay_buffer.py research/pcl_rl/replay_buffer.py +1 -1

No files found.
--- a/research/pcl_rl/replay_buffer.py
+++ b/research/pcl_rl/replay_buffer.py
@@ -133,7 +133,7 @@ class PrioritizedReplayBuffer(ReplayBuffer):
      self.remove_idx = idxs[-1] + 1 - self.init_length
    elif self.eviction_strategy == 'rank':
      # remove lowest-priority indices
-      idxs = np.argpartition(self.priorities, n)[:n]
+      idxs = np.argpartition(self.priorities, n-1)[:n]

    return idxs