add pin memory to dataloader (#750)

GiovanniCanali · web-flow · commit fa02aa656e69 · 2025-12-12T10:53:59.000+01:00
diff --git a/pina/data/data_module.py b/pina/data/data_module.py
@@ -521,6 +521,7 @@ def _create_dataloader(self, split, dataset):
                 collate_fn=collate,
                 sampler=sampler,
                 num_workers=self.num_workers,
+                pin_memory=self.pin_memory,
             )
         dataloader = DummyDataloader(dataset)
         dataloader.dataset = self._transfer_batch_to_device(

Original file line number	Diff line number	Diff line change
`@@ -521,6 +521,7 @@ def _create_dataloader(self, split, dataset):`
`521`	`521`	`collate_fn=collate,`
`522`	`522`	`sampler=sampler,`
`523`	`523`	`num_workers=self.num_workers,`
	`524`	`+ pin_memory=self.pin_memory,`
`524`	`525`	`)`
`525`	`526`	`dataloader = DummyDataloader(dataset)`
`526`	`527`	`dataloader.dataset = self._transfer_batch_to_device(`