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

feat: add dataset_version as arg

This commit is contained in:
Ashis Ghosh 2024-05-12 19:46:26 -07:00
parent 1249aee3ac
commit 60be3346d6
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -26,6 +26,7 @@ def make_dataset(
dataset = LeRobotDataset( dataset = LeRobotDataset(
cfg.dataset_repo_id, cfg.dataset_repo_id,
version=cfg.dataset_version,
split=split, split=split,
delta_timestamps=delta_timestamps, delta_timestamps=delta_timestamps,
) )

1
lerobot/configs/default.yaml
View File

@ -14,6 +14,7 @@ device: cuda # cpu
# AND for the evaluation environments. # AND for the evaluation environments.
seed: ??? seed: ???
dataset_repo_id: lerobot/pusht dataset_repo_id: lerobot/pusht
dataset_version: "v1.3"
training: training:
offline_steps: ??? offline_steps: ???