diff --git a/dataset/dataset.py b/dataset/dataset.py
index f064c404c9126088164e11b1040bdf5a93362d41..24dd482ab761b4a9ceafe9628c6d7379db0bb6f9 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