search

chr5tphr / zennit

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

Added canonizer for BasicBlock module of ResNet model #50

Closed maxdreyer closed 3 years ago

maxdreyer commented 3 years ago

Smaller ResNets such as ResNet18 or ResNet34 consist of BasicBlock modules instead of Bottleneck blocks.

Added a new AttributeCanonizer for BasicBlock canonization.

This pull request addresses issue #49.