batchsize fix2

urancon · Jan 13, 2022 · b59f651 · b59f651
1 parent 5434a00
commit b59f651
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -96,7 +96,7 @@ def set_random_seed(seed):
 train_data_loader = torch.utils.data.DataLoader(dataset=train_set,
                                                 batch_size=batchsize,
                                                 shuffle=True,
-                                                drop_last=False,
+                                                drop_last=True,
                                                 pin_memory=True)
 
 val_data_loader = torch.utils.data.DataLoader(dataset=val_set,