Merge pull request #325 from torchmd/revert-322-zero_workers

Disable persistent_workers completely, enable pin_memory in DataModule

Author: RaulPPelaez

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=False,
+            pin_memory=True,
             shuffle=shuffle,
         )