This commit is contained in:
Glenn Jocher 2019-04-15 18:19:45 +02:00
parent 00cc02ba91
commit a06abce40b
1 changed files with 1 additions and 1 deletions

View File

@ -92,7 +92,7 @@ def train(
dataloader = DataLoader(dataset,
batch_size=batch_size,
num_workers=num_workers,
shuffle=False,
shuffle=True,
pin_memory=False,
collate_fn=dataset.collate_fn,
sampler=sampler)