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

Add torchvision.ops.FrozenBatchNorm2d to types #202

Open rachtibat opened 9 months ago

rachtibat commented 9 months ago

torchvision.ops.FrozenBatchNorm2d is part of many SOTA models and contains the same attributes/buffers as the torch.nn.BatchNorm2d module. Hence, it can be canonized with the existing SequentialMergeBatchNorm class.