condition and supervise

broadinstitute · Jan 16, 2025 · 71c5b51 · 71c5b51
1 parent 2eb72ca
commit 71c5b51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py
@@ -541,7 +541,7 @@ def plot_ecgs(self, epoch=None, logs=None, num_rows=1, num_cols=4, reseed=None,
         plt.close()
 
     def plot_reconstructions(
-        self, images_original, diffusion_amount=0, epoch=None, logs=None, num_rows=4, num_cols=4, prefix='./figures/',
+        self, images_original, diffusion_amount=0, epoch=None, logs=None, num_rows=2, num_cols=2, prefix='./figures/',
     ):
         images = images_original[0][self.tensor_map.input_name()]
         self.normalizer.update_state(images)