Update train.py

This commit is contained in:
Glenn Jocher 2019-03-25 15:06:22 +01:00 committed by GitHub
parent 06d264198c
commit c16b587ac7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

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