diff --git a/train.py b/train.py index 0b3eb4e4..5bce4adc 100644 --- a/train.py +++ b/train.py @@ -92,7 +92,7 @@ def train( dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=num_workers, - shuffle=True, + shuffle=False, pin_memory=True, collate_fn=dataset.collate_fn, sampler=sampler)