Fixing FID warnings flake issue

Summary: Fixing a previous flake issue with logging a warning. Reviewed By: JKSenthil Differential Revision: D49288048 fbshipit-source-id: 311b0608ef122556e24f89a38bb861905ae429bf
pytorch · Sep 18, 2023 · ceb32b1 · ceb32b1
1 parent e0444c6
commit ceb32b1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torcheval/metrics/image/fid.py b/torcheval/metrics/image/fid.py
@@ -174,6 +174,7 @@ def compute(self: TFrechetInceptionDistance) -> Tensor:
                 f"but currently running with {self.num_real_images} real images and {self.num_fake_images} fake images."
                 "Returning 0.0",
                 RuntimeWarning,
+                stacklevel=2,
             )
 
             return torch.tensor(0.0)