From 575924ed606b9f7e9ae42656e163757e427b06ba Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 23 Mar 2022 21:40:53 -0700 Subject: [PATCH] Update test crop for new RegNet-V weights to match Y --- timm/models/regnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/regnet.py b/timm/models/regnet.py index 87ea32a6..3e22bf56 100644 --- a/timm/models/regnet.py +++ b/timm/models/regnet.py @@ -155,10 +155,10 @@ default_cfgs = dict( regnety_040s_gn=_cfg(url=''), regnetv_040=_cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/regnetv_040_ra3-c248f51f.pth', - first_conv='stem'), + first_conv='stem', crop_pct=1.0, test_input_size=(3, 288, 288)), regnetv_064=_cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/regnetv_064_ra3-530616c2.pth', - first_conv='stem'), + first_conv='stem', crop_pct=1.0, test_input_size=(3, 288, 288)), regnetz_005=_cfg(url=''), regnetz_040=_cfg(