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, )