From f30743cfe14caa5b6fa9274f7855629a4540062c Mon Sep 17 00:00:00 2001
From: Raul <raulppelaez@gmail.com>
Date: Thu, 16 May 2024 11:23:50 +0200
Subject: [PATCH] Revert "Disable persistent_workers with num_workers=0 and
 disable pin_memory in DataModule "

---
 torchmdnet/data.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torchmdnet/data.py b/torchmdnet/data.py
index e52a374a9..ba91e8bed 100644
--- a/torchmdnet/data.py
+++ b/torchmdnet/data.py
@@ -133,8 +133,8 @@ def _get_dataloader(self, dataset, stage, store_dataloader=True):
             dataset=dataset,
             batch_size=batch_size,
             num_workers=self.hparams["num_workers"],
-            persistent_workers=True if self.hparams["num_workers"] > 0 else False,
-            pin_memory=False,
+            persistent_workers=True,
+            pin_memory=True,
             shuffle=shuffle,
         )