mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ross Wightman 2024-12-02 16:55:29 -08:00 committed by Ross Wightman
parent 5fe5f9d488
commit 464885e135
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
timm/layers/conv_bn_act.py
View File

@ -82,7 +82,8 @@ class ConvNormAct(nn.Module):
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