Minor fix in PER

DLR-RM · May 24, 2024 · be00231 · be00231
1 parent f57444a
commit be00231
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stable_baselines3/common/prioritized_replay_buffer.py b/stable_baselines3/common/prioritized_replay_buffer.py
@@ -126,7 +126,7 @@ def __init__(
 
         assert optimize_memory_usage is False, "PrioritizedReplayBuffer doesn't support optimize_memory_usage=True"
 
-        self.min_priority = 1e-8
+        self.min_priority = min_priority
         self.alpha = alpha
         self.max_priority = self.min_priority  # priority for new samples, init as eps
         # Track the training progress remaining (from 1 to 0)