Skip to content

Commit 575924e

Browse files
committed
Update test crop for new RegNet-V weights to match Y
1 parent 8f6d638 commit 575924e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/regnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -155,10 +155,10 @@ def _cfg(url='', **kwargs):
155155
regnety_040s_gn=_cfg(url=''),
156156
regnetv_040=_cfg(
157157
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/regnetv_040_ra3-c248f51f.pth',
158-
first_conv='stem'),
158+
first_conv='stem', crop_pct=1.0, test_input_size=(3, 288, 288)),
159159
regnetv_064=_cfg(
160160
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/regnetv_064_ra3-530616c2.pth',
161-
first_conv='stem'),
161+
first_conv='stem', crop_pct=1.0, test_input_size=(3, 288, 288)),
162162

163163
regnetz_005=_cfg(url=''),
164164
regnetz_040=_cfg(

0 commit comments

Comments
 (0)