Skip to content

Commit b6da78f

Browse files
committed
Fix WSquareRule on Dense layers
1 parent a83e276 commit b6da78f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lrp/rules.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -433,7 +433,7 @@ for R in (ZeroRule, EpsilonRule)
433433
end
434434

435435
# Fast implementation for Dense layer using Tullio.jl's einsum notation:
436-
for R in (ZeroRule, EpsilonRule, GammaRule, WSquareRule)
436+
for R in (ZeroRule, EpsilonRule, GammaRule)
437437
@eval function lrp!(Rₖ, rule::$R, layer::Dense, aₖ, Rₖ₊₁)
438438
reset! = get_layer_resetter(rule, layer)
439439
modify_layer!(rule, layer)

0 commit comments

Comments
 (0)