Set persistent_workers to False

torchmd · May 16, 2024 · e3924d7 · e3924d7
1 parent f30743c
commit e3924d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         )