small adjustment to not risk video loader segmentation fault

2024-05-02 19:33:01 +00:00 · 2024-05-02 19:33:01 +00:00 · 38cf61e1a5
parent c6693e296b
commit 38cf61e1a5
1 changed files with 1 additions and 1 deletions
--- a/examples/1_load_lerobot_dataset.py
+++ b/examples/1_load_lerobot_dataset.py
@ -80,7 +80,7 @@ print(f"{dataset[0]['action'].shape=}")  # (64,c)
 # because they are just PyTorch datasets.
 dataloader = torch.utils.data.DataLoader(
    dataset,
-    num_workers=0,
+    num_workers=1,
    batch_size=32,
    shuffle=True,
 )