update comment

2025-03-14 16:59:31 +01:00 · 2025-03-14 16:59:31 +01:00 · 4bdbf2f6e0
parent 4e9b4dd380
commit 4bdbf2f6e0
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/sampler.py
+++ b/lerobot/common/datasets/sampler.py
@ -140,8 +140,8 @@ class PrioritizedSampler(Sampler[int]):
        Args:
            data_len: Total number of samples in the dataset.
            alpha: Exponent for priority scaling. Default is 0.6.
+            beta: Used in important sampling (IS). Default is 0.4,
            eps: Small constant to avoid zero priorities.
-            replacement: Whether to sample with replacement.
            num_samples_per_epoch: Number of samples per epoch (default is data_len).
        """
        self.data_len = data_len