chr5tphr / zennit

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

Core: Warn the user when falling back to gradient #147

Open chr5tphr opened 2 years ago

chr5tphr commented 2 years ago

Some modules are implicitly mapped to the gradient.

We can explicitly map Module types to None in their respective module_map in composites and warn the user when no rule is found to prevent hidden incompatibilities.

An example for an implicitly mapped module is MaxPooling.