From 8d9ebe3788b5826d4bde5595eb414ffc260fe6e6 Mon Sep 17 00:00:00 2001
From: lixinjie <lixinjie@meituan.com>
Date: Mon, 31 May 2021 00:36:13 +0800
Subject: [PATCH] fix some bugs

---
 timm/models/twins.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/timm/models/twins.py b/timm/models/twins.py
index eded603a..c69bcf9b 100644
--- a/timm/models/twins.py
+++ b/timm/models/twins.py
@@ -310,9 +310,7 @@ class Twins(nn.Module):
                     else:
                         self.patch_embeds.append(
                             PatchEmbed(img_size // patch_size // s, strides[i-1], embed_dims[i - 1], embed_dims[i]))
-                    s = s * strides[i-1]        
-        if self.task == 'cls':
-            del self.norm_list
+                    s = s * strides[i-1]
 
         self.blocks = nn.ModuleList()
         dpr = [x.item() for x in torch.linspace(0, drop_path_rate, sum(depths))]  # stochastic depth decay rule