diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index 3f162c79..aab44365 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -573,9 +573,8 @@ class RepVggBlock(nn.Module): if isinstance(m, nn.BatchNorm2d): nn.init.normal_(m.weight, .1, .1) nn.init.normal_(m.bias, 0, .1) - for attn in (self.attn, self.attn_last): - if hasattr(attn, 'reset_parameters'): - attn.reset_parameters() + if hasattr(self.attn, 'reset_parameters'): + self.attn.reset_parameters() def forward(self, x): if self.identity is None: diff --git a/timm/models/layers/eca.py b/timm/models/layers/eca.py index d0d8f74a..f2980730 100644 --- a/timm/models/layers/eca.py +++ b/timm/models/layers/eca.py @@ -110,4 +110,4 @@ class CecaModule(nn.Module): return x * y.expand_as(x) -CircularEfficientChannelAttn = CecaModule \ No newline at end of file +CircularEfficientChannelAttn = CecaModule