tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update doc

Browse Source
This commit is contained in:
Simon Alibert
2024-10-31 14:08:13 +01:00
parent e1845d4dcc
commit c70b8d0abc
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -108,7 +108,8 @@ class LeRobotDataset(torch.utils.data.Dataset):
- tasks contains the prompts for each task of the dataset, which can be used for - tasks contains the prompts for each task of the dataset, which can be used for
task-conditionned training. task-conditionned training.
- hf_dataset (from datasets.Dataset), which will read any values from parquet files. - hf_dataset (from datasets.Dataset), which will read any values from parquet files.
- (optional) videos from which frames are loaded to be synchronous with data from parquet files. - videos (optional) from which frames are loaded to be synchronous with data from parquet files.
A typical LeRobotDataset looks like this from its root path: A typical LeRobotDataset looks like this from its root path:
. .
├── data ├── data
@@ -128,7 +129,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
│ ├── info.json │ ├── info.json
│ ├── stats.json │ ├── stats.json
│ └── tasks.jsonl │ └── tasks.jsonl
└── videos (optional) └── videos
├── chunk-000 ├── chunk-000
│ ├── observation.images.laptop │ ├── observation.images.laptop
│ │ ├── episode_000000.mp4 │ │ ├── episode_000000.mp4