corrected noise prior

flatironinstitute · Jul 3, 2023 · 4ccbe60 · 4ccbe60
1 parent 65eb909
commit 4ccbe60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cryo_sbi/wpa_simulator/noise.py b/src/cryo_sbi/wpa_simulator/noise.py
@@ -36,7 +36,7 @@ def get_snr(images, snr):
     )
     signal_power = images[:, mask].pow(2).mean().sqrt()  # torch.std(image[mask])
 
-    noise_power = signal_power / torch.sqrt(snr.to(images.device))
+    noise_power = signal_power / torch.sqrt(torch.pow(10, snr))
 
     return noise_power