From d2dc37e03ba19294afd07bd1f0af41465f226df4 Mon Sep 17 00:00:00 2001
From: lcalmettes <leo.calmettes@etu.ec-lyon.fr>
Date: Mon, 12 May 2025 16:06:33 +0200
Subject: [PATCH] =?UTF-8?q?=09modifi=C3=A9=C2=A0:=20=20=20=20=20=20=20=20?=
=?UTF-8?q?=20dataset/dataset.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
dataset/dataset.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/dataset/dataset.py b/dataset/dataset.py
index 0ea5107..f064c40 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -252,7 +252,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
dataset=train_dataset,
batch_size=batch_size,
shuffle=shuffle,
- num_workers=1,
+ num_workers=8,
collate_fn=None,
pin_memory=False,
)
@@ -260,7 +260,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
dataset=val_dataset,
batch_size=batch_size,
shuffle=shuffle,
- num_workers=1,
+ num_workers=8,
collate_fn=None,
pin_memory=False,
)
--
GitLab