drop last

Author: r9y9

train.py

@@ -963,7 +963,7 @@ def restore_parts(path, model):
     data_loader = data_utils.DataLoader(
         dataset, batch_size=hparams.batch_size,
         num_workers=hparams.num_workers, sampler=sampler,
-        collate_fn=collate_fn, pin_memory=hparams.pin_memory)
+        collate_fn=collate_fn, pin_memory=hparams.pin_memory, drop_last=True)
 
     device = torch.device("cuda" if use_cuda else "cpu")