From f615474be317b1e015c082b7dabd391f461c10b7 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Thu, 27 May 2021 18:12:22 -0700
Subject: [PATCH] Fix broken test, repvgg block doesn't have attn_last attr.

---
 timm/models/byobnet.py    | 5 ++---
 timm/models/layers/eca.py | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py
index 3f162c7..aab4436 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 d0d8f74..f298073 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
-- 
GitLab