modifié : dataset/dataset.py

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