Skip to content

Commit

Permalink
1.0 instead of 1
Browse files Browse the repository at this point in the history
  • Loading branch information
chiheem committed Oct 11, 2024
1 parent d068dcc commit f65d538
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -304,7 +304,7 @@ def lpips_similarity(self, generated_image: Image.Image, reference_image: Image.
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=UserWarning)
# https://lightning.ai/docs/torchmetrics/stable/image/learned_perceptual_image_patch_similarity.html
self._lpips_metric = 1 - LearnedPerceptualImagePatchSimilarity(net_type="vgg", normalize=True).to(
self._lpips_metric = 1.0 - LearnedPerceptualImagePatchSimilarity(net_type="vgg", normalize=True).to(
self._device
)

Expand Down

0 comments on commit f65d538

Please sign in to comment.