yacwc

2021-07-12 13:00:21 -04:00
parent f46d193826
commit 3e54faeb44
2 changed files with 4 additions and 3 deletions
--- a/train.py
+++ b/train.py
@@ -45,7 +45,7 @@ def run():

    train_data_loader = torch.utils.data.DataLoader(
        train_dataset,
-        batch_size=8,
+        batch_size=16,
        shuffle=True,
        num_workers=4,
        collate_fn=utils.collate_fn,
@@ -53,7 +53,7 @@ def run():

    val_data_loader = torch.utils.data.DataLoader(
        val_dataset,
-        batch_size=8,
+        batch_size=16,
        shuffle=True,
        num_workers=4,
        collate_fn=utils.collate_fn,