diff --git a/src/ptbench/data/augmentations.py b/src/ptbench/data/augmentations.py index 1f4adc601b665287ba7f605b39179ea0028ab070..a104ec4ac26b1540ed2a94a0509563fbeff0b4f8 100644 --- a/src/ptbench/data/augmentations.py +++ b/src/ptbench/data/augmentations.py @@ -287,7 +287,7 @@ class ElasticDeformation: logger.info( f"Applying data-augmentation using {instances} processes..." ) - self._mp_pool = multiprocessing.pool.Pool(instances) + self._mp_pool = multiprocessing.get_context("spawn").Pool(instances) else: self._mp_pool = None @@ -296,7 +296,9 @@ class ElasticDeformation: if self._mp_pool is None and self._parallel == -2: # auto-tunning on first batch instances = min(img.shape[0], multiprocessing.cpu_count()) - self._mp_pool = multiprocessing.pool.Pool(instances) + self._mp_pool = multiprocessing.get_context("spawn").Pool( + instances + ) return _elastic_deformation_on_batch( img,