diff --git a/qim3d/utils/data.py b/qim3d/utils/data.py index 332856e43248ffc083059ac93276697625a48de3..fb2a03845870beabe1b1c23970c451a3750ba67e 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.