This commit is contained in:
Cadene 2024-05-06 00:44:20 +00:00
parent 5a867e34af
commit cf4b4c5a18
1 changed files with 1 additions and 1 deletions

View File

@ -80,7 +80,7 @@ print(f"{dataset[0]['action'].shape=}\n") # (64,c)
# PyTorch datasets.
dataloader = torch.utils.data.DataLoader(
dataset,
num_workers=0,
num_workers=1,
batch_size=32,
shuffle=True,
)