From bde9554604061e580f8dcd54c06ca2b811696e3c Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sun, 8 Jan 2023 12:30:10 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index fb950f61..f909b079 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -596,7 +596,7 @@ DOWNSAMPLE_LAYERS_FOUR_STAGES = [partial(Downsampling, )] + \ [partial(Downsampling, kernel_size=3, stride=2, padding=1, - pre_norm=partial(LayerNormGeneral, bias=False, eps=1e-6), pre_permute=True + pre_norm=partial(LayerNormGeneral, bias=False, eps=1e-6), pre_permute=False )]*3