Update lerobot/common/datasets/lerobot_dataset.py

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Jade Choghari
2025-03-04 13:27:40 +03:00
committed by GitHub
parent e1732b4954
commit 744906098a

View File

@@ -463,7 +463,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
video files are already present on local disk, they won't be downloaded again. Defaults to
True.
video_backend (str | None, optional): Video backend to use for decoding videos. Defaults to torchcodec.
You can also use the 'pyav' decoder used by Torchvision.
You can also use the 'pyav' decoder used by Torchvision, which used to be the default option, or 'video_reader' which is another decoder of Torchvision.
"""
super().__init__()
self.repo_id = repo_id