issues
search
chr5tphr
/
zennit
Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.
Other
190
stars
32
forks
source link
Fix MergeBatchNorm for all Convolutions
#71
Closed
chr5tphr
closed
2 years ago
chr5tphr
commented
2 years ago
previously, MergeBatchNorm only worked correctly for Linear and Conv2d
for ConvTranspose, the BatchNorm's scale was falsely attempted to be merged into the input channel dimension
the
unsqueezing
dimensions were fixed for 1d/3d inputs