search

chr5tphr / zennit

Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.
Other
183 stars 33 forks source link

remove nn.MultiheadAttention from Activation types.py #191

Closed rachtibat closed 1 year ago

rachtibat commented 1 year ago

Bugfix: Zennit assigns per default the Pass rule to the nn.MultiheadAttention layer. However, the MultiheadAttention module requires another rule.

chr5tphr commented 1 year ago

Thanks a lot!