From 56c285445cafa3758a631e36c0f209efc4c82ccf Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 24 Aug 2023 14:16:21 -0700 Subject: [PATCH] Wrong pool size for 384x384 inception_next_base --- timm/models/inception_next.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/inception_next.py b/timm/models/inception_next.py index da3a582bdf..d22593cb56 100644 --- a/timm/models/inception_next.py +++ b/timm/models/inception_next.py @@ -399,7 +399,7 @@ def _cfg(url='', **kwargs): 'inception_next_base.sail_in1k_384': _cfg( hf_hub_id='timm/', # url='https://github.com/sail-sg/inceptionnext/releases/download/model/inceptionnext_base_384.pth', - input_size=(3, 384, 384), pool_size=(10, 10), crop_pct=1.0, + input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, ), })