From 01e16fd7cd77dd9976fd32997a7c92ccbc079462 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Thu, 14 Dec 2023 15:06:48 +0100
Subject: [PATCH] [data.augmentations] Temporarily disables within-batch
parallelisation of elastic deformation (mitigates #53)
---
src/ptbench/data/augmentations.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/data/augmentations.py b/src/ptbench/data/augmentations.py
index a104ec4a..f890c530 100644
--- a/src/ptbench/data/augmentations.py
+++ b/src/ptbench/data/augmentations.py
@@ -257,7 +257,7 @@ class ElasticDeformation:
spline_order: int = 1,
mode: str = "nearest",
p: float = 1.0,
- parallel: int = -2,
+ parallel: int = -1,
):
self.alpha: float = alpha
self.sigma: float = sigma
--
GitLab