dists: Fix sigmoid call

braun-steven · Oct 23, 2023 · 41c0fc2 · 41c0fc2
1 parent 9ff96c3
commit 41c0fc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simple_einet/layers/distributions/bernoulli.py b/simple_einet/layers/distributions/bernoulli.py
@@ -28,4 +28,4 @@ def __init__(self, num_features: int, num_channels: int, num_leaves: int, num_re
 
     def _get_base_distribution(self):
         # Use sigmoid to ensure, that probs are in valid range
-        return dist.Bernoulli(probs=self.sigmoid(self.probs))
+        return dist.Bernoulli(probs=torch.sigmoid(self.probs))