diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 9bb85c44..e694fd67 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -228,7 +228,7 @@ class Downsampling(nn.Module): """ def __init__(self, in_channels, out_channels, kernel_size, stride=1, padding=0, - pre_norm=None, post_norm=None): + pre_norm=None, post_norm=None, pre_permute = False): super().__init__() self.pre_norm = pre_norm(in_channels) if pre_norm else nn.Identity() self.pre_permute = pre_permute