drop last = true

2026-09-09 11:16:00 +08:00 · 2018-04-24 12:52:43 -07:00
parent 30c6ac8e71
commit 85bbab74b3
1 changed files with 1 additions and 1 deletions
@@ -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),