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
@ -311,8 +311,6 @@ class Twins(nn.Module):
                        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

        self.blocks = nn.ModuleList()
        dpr = [x.item() for x in torch.linspace(0, drop_path_rate, sum(depths))]  # stochastic depth decay rule