Further revert clamp changes (broke tests)

bayesflow-org · Nov 21, 2024 · 7906c1e · 7906c1e
1 parent 433226b
commit 7906c1e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bayesflow/networks/coupling_flow/transforms/affine_transform.py b/bayesflow/networks/coupling_flow/transforms/affine_transform.py
@@ -11,6 +11,7 @@ class AffineTransform(Transform):
     def __init__(self, clamp: bool = True, **kwargs):
         super().__init__(**kwargs)
         self.clamp = clamp
+        self.clamp_factor = 3.0
 
     @property
     def params_per_dim(self):
@@ -29,7 +30,7 @@ def constrain_parameters(self, parameters: dict[str, Tensor]) -> dict[str, Tenso
 
         # soft clamp
         if self.clamp:
-            scale = ops.arcsinh(scale)
+            scale = self.clamp_factor * ops.arcsinh(scale)
 
         parameters["scale"] = scale
         return parameters