Reduce LRS epsilon for lower reconstruction error

davidnabergoj · Nov 14, 2023 · fc490fc · fc490fc
1 parent fdd4390
commit fc490fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/normalizing_flows/bijections/finite/autoregressive/transformers/spline/linear_rational.py b/normalizing_flows/bijections/finite/autoregressive/transformers/spline/linear_rational.py
@@ -20,7 +20,7 @@ def __init__(self, event_shape: Union[torch.Size, Tuple[int, ...]], boundary: fl
         self.min_bin_height = 1e-2
         self.min_d = 1e-5
         self.const = math.log(math.exp(1 - self.min_d) - 1)  # to ensure identity initialization
-        self.eps = 1e-7  # Epsilon for numerical stability when computing forward/inverse
+        self.eps = 5e-10  # Epsilon for numerical stability when computing forward/inverse
 
     @property
     def n_parameters(self) -> int: