Set persistent_workers to True in test dataloader (#164)

creatis-myriad · Jun 8, 2024 · 4ceef9b · 4ceef9b
1 parent 412c1d9
commit 4ceef9b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ascent/datamodules/nnunet_datamodule.py b/ascent/datamodules/nnunet_datamodule.py
@@ -281,6 +281,7 @@ def test_dataloader(self) -> DataLoader:  # noqa: D102
             num_workers=self.hparams.num_workers,
             pin_memory=self.hparams.pin_memory,
             shuffle=False,
+            persistent_workers=True,
         )
 
     def get_train_val_test_loading_transforms(self) -> tuple[list[Callable], ...]: