See if we can avoid some model / layer pickle issues with the aa attr…

… in ConvNormAct
huggingface · Dec 3, 2024 · 464885e · 464885e
1 parent 5fe5f9d
commit 464885e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/timm/layers/conv_bn_act.py b/timm/layers/conv_bn_act.py
@@ -82,7 +82,8 @@ def out_channels(self):
     def forward(self, x):
         x = self.conv(x)
         x = self.bn(x)
-        if self.aa is not None:
+        aa = getattr(self, 'aa', None)
+        if aa is not None:
             x = self.aa(x)
         return x