chr5tphr / zennit

Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ResNet canonizer misses canonizer for BasicBlock module

maxdreyer opened this issue · comments

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

In order to ensure canonization of small ResNets, a BasicBlock should be added to the ResNetCanonizer.