From 7cfa5bea7c344ab50a79cf2b39ae7a12630f28d5 Mon Sep 17 00:00:00 2001 From: Stephen Farr Date: Thu, 16 May 2024 10:06:49 +0200 Subject: [PATCH 1/2] enable num_workers=0 --- torchmdnet/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchmdnet/data.py b/torchmdnet/data.py index ba91e8bed..b587f0a12 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=True if self.hparams["num_workers"] > 0 else False, pin_memory=True, shuffle=shuffle, ) From fe3036a3c1da1809bc444b629053e1b96b6bb41c Mon Sep 17 00:00:00 2001 From: Raul Date: Thu, 16 May 2024 10:45:16 +0200 Subject: [PATCH 2/2] Set pin_memory to false --- torchmdnet/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchmdnet/data.py b/torchmdnet/data.py index b587f0a12..e52a374a9 100644 --- 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, )