bugfix spike_slab log_prob

rctn · Jun 1, 2022 · 88aad94 · 88aad94
1 parent c48b2c3
commit 88aad94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparsecoding/priors/spike_slab.py b/sparsecoding/priors/spike_slab.py
@@ -91,7 +91,7 @@ def log_prob(
         log_prob[slab_mask] = (
             torch.log(torch.tensor(1. - self.p_spike))
             - torch.log(torch.tensor(self.scale))
-            - sample[slab_mask] / self.scale
+            - torch.abs(sample[slab_mask]) / self.scale
         )
         if self.positive_only:
             log_prob[sample < 0.] = -torch.inf