update comment

This commit is contained in:
Pepijn 2025-03-14 16:59:31 +01:00
parent 4e9b4dd380
commit 4bdbf2f6e0
1 changed files with 1 additions and 1 deletions

View File

@ -140,8 +140,8 @@ class PrioritizedSampler(Sampler[int]):
Args: Args:
data_len: Total number of samples in the dataset. data_len: Total number of samples in the dataset.
alpha: Exponent for priority scaling. Default is 0.6. 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. 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). num_samples_per_epoch: Number of samples per epoch (default is data_len).
""" """
self.data_len = data_len self.data_len = data_len