Update metaformers.py

2 years ago · 926d886527
parent 358c4ae7ec
commit 926d886527
1 changed files with 1 additions and 1 deletions
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@ -527,7 +527,7 @@ class MetaFormerBlock(nn.Module):

        super().__init__()
        
-        self.downsample = nn.Identity()
+        self.downsample = downsample
        
        self.norm1 = norm_layer(dim)
        self.token_mixer = token_mixer(dim=dim, drop=drop)