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
2 changed files with 2 additions and 2 deletions

View File

@@ -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,
)