drop last = true

2018-04-24 12:52:43 -07:00 · 2018-04-24 12:52:43 -07:00 · d3233d0503
parent 0f721090b3
commit d3233d0503
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -333,7 +333,7 @@ def main(args):

    train_loader = DataLoader(train_dataset, batch_size=c.batch_size,
                              shuffle=False, collate_fn=train_dataset.collate_fn,
-                              drop_last=False, num_workers=c.num_loader_workers,
+                              drop_last=True, num_workers=c.num_loader_workers,
                              pin_memory=True)

    val_dataset = Dataset(os.path.join(c.data_path, c.meta_file_val),