added comment for torch no grad

flatironinstitute · May 21, 2024 · bf2c319 · bf2c319
1 parent 9e8e9d7
commit bf2c319
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cryo_sbi/wpa_simulator/cryo_em_simulator.py b/src/cryo_sbi/wpa_simulator/cryo_em_simulator.py
@@ -12,7 +12,7 @@
 from cryo_sbi.utils.micrograph_utils import RandomMicrographPatches
 from cryo_sbi.utils.image_utils import NormalizeIndividual
 
-
+#@torch.no_grad()
 def cryo_em_simulator(
     models,
     index,