From e3924d7e19a0f6bfa65caa4929da5dbdbe1854f5 Mon Sep 17 00:00:00 2001
From: Raul <raulppelaez@gmail.com>
Date: Thu, 16 May 2024 11:44:28 +0200
Subject: [PATCH] Set persistent_workers to False

---
 torchmdnet/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchmdnet/data.py b/torchmdnet/data.py
index ba91e8be..986e19f7 100644
--- a/torchmdnet/data.py
+++ b/torchmdnet/data.py
@@ -133,7 +133,7 @@ def _get_dataloader(self, dataset, stage, store_dataloader=True):
             dataset=dataset,
             batch_size=batch_size,
             num_workers=self.hparams["num_workers"],
-            persistent_workers=True,
+            persistent_workers=False,
             pin_memory=True,
             shuffle=shuffle,
         )