test with LSM prior

rctn · May 25, 2022 · c48b2c3 · c48b2c3
1 parent 6ca4f1e
commit c48b2c3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/tests/inference/common.py b/tests/inference/common.py
@@ -1,6 +1,7 @@
 import torch
 
 from sparsecoding.priors.l0 import L0Prior
+from sparsecoding.priors.lsm import LSMPrior
 from sparsecoding.priors.spike_slab import SpikeSlabPrior
 from sparsecoding.data.datasets.bars import BarsDataset
 
@@ -24,6 +25,12 @@
             ).type(torch.float32)
         ),
     ),
+    LSMPrior(
+        dim=2 * PATCH_SIZE,
+        alpha=80.0,
+        beta=0.02,
+        positive_only=False,
+    ),
 ]
 
 DATASET = [

diff --git a/tests/inference/test_LSM.py b/tests/inference/test_LSM.py
@@ -28,7 +28,7 @@ def test_inference(self):
 
             a = inference_method.infer(data, DICTIONARY)
 
-            self.assertAllClose(a, dataset.weights, atol=5e-2)
+            self.assertAllClose(a, dataset.weights, atol=7.5e-2)
 
 
 if __name__ == "__main__":