diff --git a/examples/1_load_hugging_face_dataset.py b/examples/1_load_hugging_face_dataset.py index 35f3c685..17d28914 100644 --- a/examples/1_load_hugging_face_dataset.py +++ b/examples/1_load_hugging_face_dataset.py @@ -37,7 +37,7 @@ from datasets import load_dataset # TODO(rcadene): list available datasets on lerobot page using `datasets` # download/load hugging face dataset in pyarrow format -hf_dataset, fps = load_dataset("lerobot/pusht", split="train"), 10 +hf_dataset, fps = load_dataset("lerobot/pusht", revision="v1.0", split="train"), 10 # display name of dataset and its features print(f"{hf_dataset=}") diff --git a/examples/2_load_lerobot_dataset.py b/examples/2_load_lerobot_dataset.py index ace9c174..49a53d8e 100644 --- a/examples/2_load_lerobot_dataset.py +++ b/examples/2_load_lerobot_dataset.py @@ -87,7 +87,7 @@ print(f"{dataset[0]['action'].shape=}") # (64,c) # because they are just PyTorch datasets. dataloader = torch.utils.data.DataLoader( dataset, - num_workers=4, + num_workers=0, batch_size=32, shuffle=True, )