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

Add video_backend arg in MultiLeRobotDataset

This commit is contained in:
Simon Alibert 2024-05-30 17:42:01 +00:00
parent af19a89614
commit c86670f97e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/datasets/factory.py
View File

@ -74,7 +74,10 @@ def make_dataset(cfg, split: str = "train") -> LeRobotDataset | MultiLeRobotData
) )
else: else:
dataset = MultiLeRobotDataset( dataset = MultiLeRobotDataset(
cfg.dataset_repo_id, split=split, delta_timestamps=cfg.training.get("delta_timestamps") cfg.dataset_repo_id,
split=split,
delta_timestamps=cfg.training.get("delta_timestamps"),
video_backend=cfg.video_backend,
) )
if cfg.get("override_dataset_stats"): if cfg.get("override_dataset_stats"):