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

Add torchvision.ops.FrozenBatchNorm2d to types #202

Open rachtibat opened 1 year ago

rachtibat commented 1 year 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.