Set pin_memory to false

torchmd · May 16, 2024 · fe3036a · fe3036a
1 parent 7cfa5be
commit fe3036a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmdnet/data.py b/torchmdnet/data.py
@@ -134,7 +134,7 @@ def _get_dataloader(self, dataset, stage, store_dataloader=True):
             batch_size=batch_size,
             num_workers=self.hparams["num_workers"],
             persistent_workers=True if self.hparams["num_workers"] > 0 else False,
-            pin_memory=True,
+            pin_memory=False,
             shuffle=shuffle,
         )