fix some bugs

4 years ago · 8d9ebe3788
parent bd996241f6
commit 8d9ebe3788
1 changed files with 1 additions and 3 deletions
--- 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