compute inception score in test steps

broadinstitute · Jan 20, 2025 · 4b2a8f4 · 4b2a8f4
1 parent f064d9a
commit 4b2a8f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/models/diffusion_blocks.py b/ml4h/models/diffusion_blocks.py
@@ -886,7 +886,7 @@ def test_step(self, batch):
                 num_images=self.batch_size, diffusion_steps=20
             )
             self.kid.update_state(images, generated_images)
-            self.inception_score.update_state(images, generated_images)
+            self.inception_score.update_state(generated_images)
 
         return {m.name: m.result() for m in self.metrics}