wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve docstring

This commit is contained in:
Simon Alibert 2024-06-10 13:46:53 +00:00
parent a65b67ff23
commit f57090fd4c
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lerobot/common/datasets/transforms.py
View File

@ -13,11 +13,11 @@ class RandomSubsetApply(Transform):
Args:
transforms (sequence or torch.nn.Module): list of transformations
p (list of floats or None, optional): probability of each transform being picked.
If ``p`` doesn't sum to 1, it is automatically normalized. If ``None``
(default), all transforms have the same probability.
p (list of floats or None, optional): represents the multinomial probabilities
(with no replacement) used for sampling the transform. If the sum of the weights is not 1, they
will be normalized. If ``None`` (default), all transforms have the same probability.
n_subset (int or None): number of transformations to apply. If ``None``,
all transforms are applied.
all transforms are applied. Must be in [1, len(transforms)].
random_order (bool): apply transformations in a random order
"""