From 94a5215ade3737e47c27b34d8477eac1e1e213f6 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 22 Jan 2024 15:13:49 +0100
Subject: [PATCH] [data.augmentations] Re-enable within-batch parallelisation
of elastic deformation
---
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 f890c530..a104ec4a 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 = -1,
+ parallel: int = -2,
):
self.alpha: float = alpha
self.sigma: float = sigma
--
GitLab