update comment
This commit is contained in:
parent
4e9b4dd380
commit
4bdbf2f6e0
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue