Skip to content
Snippets Groups Projects
Commit 01e16fd7 authored by André Anjos's avatar André Anjos :speech_balloon: Committed by Daniel CARRON
Browse files

[data.augmentations] Temporarily disables within-batch parallelisation of...

[data.augmentations] Temporarily disables within-batch parallelisation of elastic deformation (mitigates #53)
parent 0152c9c3
Branches
Tags
1 merge request!12Adds grad-cam support on classifiers
...@@ -257,7 +257,7 @@ class ElasticDeformation: ...@@ -257,7 +257,7 @@ class ElasticDeformation:
spline_order: int = 1, spline_order: int = 1,
mode: str = "nearest", mode: str = "nearest",
p: float = 1.0, p: float = 1.0,
parallel: int = -2, parallel: int = -1,
): ):
self.alpha: float = alpha self.alpha: float = alpha
self.sigma: float = sigma self.sigma: float = sigma
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment