From c70b8d0abc7a1d1e97fc2b4aa1be913cb17ae9ea Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Thu, 31 Oct 2024 14:08:13 +0100 Subject: [PATCH] Update doc --- lerobot/common/datasets/lerobot_dataset.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lerobot/common/datasets/lerobot_dataset.py b/lerobot/common/datasets/lerobot_dataset.py index d09e28af..49627b16 100644 --- a/lerobot/common/datasets/lerobot_dataset.py +++ b/lerobot/common/datasets/lerobot_dataset.py @@ -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 task-conditionned training. - 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: . ├── data @@ -128,7 +129,7 @@ class LeRobotDataset(torch.utils.data.Dataset): │ ├── info.json │ ├── stats.json │ └── tasks.jsonl - └── videos (optional) + └── videos ├── chunk-000 │ ├── observation.images.laptop │ │ ├── episode_000000.mp4