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

Hotfix test_examples.py (#87)

This commit is contained in:
Simon Alibert 2024-04-19 12:36:04 +02:00 committed by GitHub
parent 8d980940a2
commit ac0ab27333
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/1_load_hugging_face_dataset.py
View File

@ -37,7 +37,7 @@ from datasets import load_dataset
# TODO(rcadene): list available datasets on lerobot page using `datasets` # TODO(rcadene): list available datasets on lerobot page using `datasets`
# download/load hugging face dataset in pyarrow format # 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 # display name of dataset and its features
print(f"{hf_dataset=}") print(f"{hf_dataset=}")

2
examples/2_load_lerobot_dataset.py
View File

@ -87,7 +87,7 @@ print(f"{dataset[0]['action'].shape=}") # (64,c)
# because they are just PyTorch datasets. # because they are just PyTorch datasets.
dataloader = torch.utils.data.DataLoader( dataloader = torch.utils.data.DataLoader(
dataset, dataset,
num_workers=4, num_workers=0,
batch_size=32, batch_size=32,
shuffle=True, shuffle=True,
) )