From 26a8e481a5cb2a32004a796bc25c6800cd2fb7b7 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Wed, 1 Feb 2023 07:42:07 -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 aa453b6f..4c15a308 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -684,7 +684,7 @@ default_cfgs = generate_default_cfgs({ classifier='head.fc.fc2'), 'convformer_s18.sail_in1k_384': _cfg( url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18_384.pth', - classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(14,14)), + classifier='head.fc.fc2', input_size=(3, 384, 384), pool_size=(12,12)), 'convformer_s18.sail_in22k_ft_in1k': _cfg( url='https://huggingface.co/sail/dl/resolve/main/convformer/convformer_s18_in21ft1k.pth', classifier='head.fc.fc2'),