Skip to content

Commit 3f7a0b7

Browse files
NetoPedrovivekmig
authored andcommitted
Fixed error on LayerActivation documentation sample
1 parent b1a9830 commit 3f7a0b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

captum/attr/_core/layer/layer_activation.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def attribute(
112112
>>> input = torch.randn(2, 3, 32, 32, requires_grad=True)
113113
>>> # Computes layer activation.
114114
>>> # attribution is layer output, with size Nx12x32x32
115-
>>> attribution = layer_cond.attribute(input)
115+
>>> attribution = layer_act.attribute(input)
116116
"""
117117
with torch.no_grad():
118118
layer_eval = _forward_layer_eval(

0 commit comments

Comments
 (0)