From 753d551447baf19903ffbb5a5e51492f05d34a8b Mon Sep 17 00:00:00 2001
From: lcalmettes <leo.calmettes@etu.ec-lyon.fr>
Date: Mon, 12 May 2025 16:36:40 +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 f064c40..24dd482 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -254,7 +254,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
shuffle=shuffle,
num_workers=8,
collate_fn=None,
- pin_memory=False,
+ pin_memory=True,
)
data_loader_test = data.DataLoader(
dataset=val_dataset,
@@ -262,7 +262,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
shuffle=shuffle,
num_workers=8,
collate_fn=None,
- pin_memory=False,
+ pin_memory=True,
)
return data_loader_train, data_loader_test
--
GitLab