fix
This commit is contained in:
parent
5a867e34af
commit
cf4b4c5a18
|
@ -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,
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue