Skip to content

Commit 95dbca3

Browse files
authored
Fix TotalVariation citation
1 parent 36a67e7 commit 95dbca3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

captum/optim/_core/loss.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,8 @@ def __call__(self, targets_to_values: ModuleOutputMapping) -> torch.Tensor:
9494
class TotalVariation(Loss):
9595
"""
9696
Total variation denoising penalty for activations.
97-
See Simonyan, et al., 2014.
97+
See Mahendran, V. 2014. Understanding Deep Image Representations by Inverting Them.
98+
https://arxiv.org/abs/1412.0035
9899
"""
99100

100101
def _call__(self, targets_to_values: ModuleOutputMapping) -> torch.Tensor:

0 commit comments

Comments
 (0)