From a695f8d002ef63299123634823ca05b720a5a79d Mon Sep 17 00:00:00 2001
From: s184364 <s184364@student.dtu.dk>
Date: Mon, 18 Dec 2023 16:35:58 +0100
Subject: [PATCH] changed pin_workers to 0 by default: faster dataloader
---
qim3d/utils/data.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/qim3d/utils/data.py b/qim3d/utils/data.py
index 332856e4..fb2a0384 100644
--- a/qim3d/utils/data.py
+++ b/qim3d/utils/data.py
@@ -174,7 +174,7 @@ def prepare_datasets(path: str, val_fraction: float, model, augmentation):
return train_set, val_set, test_set
-def prepare_dataloaders(train_set, val_set, test_set, batch_size, shuffle_train = True, num_workers = 8, pin_memory = False):
+def prepare_dataloaders(train_set, val_set, test_set, batch_size, shuffle_train = True, num_workers = 0, pin_memory = False):
"""
Prepares the dataloaders for model training.
--
GitLab