Skip to content

Commit

Permalink
Update lpip.py
Browse files Browse the repository at this point in the history
Fix E501 Line too long (148 > 120 characters)
  • Loading branch information
Viola-Siemens authored Sep 22, 2023
1 parent 8064628 commit ea6918c
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/torchmetrics/image/lpip.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,8 @@ class LearnedPerceptualImagePatchSimilarity(Metric):
- ``img1`` (:class:`~torch.Tensor`): tensor with images of shape ``(N, 3, H, W)``
- ``img2`` (:class:`~torch.Tensor`): tensor with images of shape ``(N, 3, H, W)``
- ``detach`` (:class:`~bool`): bool indicating if return tensor if detached. Set to True if you want to save memory and do not need to backward.
- ``detach`` (:class:`~bool`): bool indicating if return tensor if detached.
Set to True if you want to save memory and do not need to backward.
As output of `forward` and `compute` the metric returns the following output
Expand Down

0 comments on commit ea6918c

Please sign in to comment.