From 57fca2b5b23c0738280dbc7e4045887d7e4fa4a4 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 10 Dec 2021 20:23:17 -0800 Subject: [PATCH] Fix c16_evos stem / first conv setup --- timm/models/byobnet.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index a2dc6153..d6e50cc8 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -157,11 +157,10 @@ default_cfgs = { 'regnetz_b16_evos': _cfgr( url='', mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), - input_size=(3, 224, 224), pool_size=(7, 7), test_input_size=(3, 288, 288), first_conv='stem.conv', - crop_pct=0.94), + input_size=(3, 224, 224), pool_size=(7, 7), test_input_size=(3, 288, 288), first_conv='stem.conv', crop_pct=0.94), 'regnetz_c16_evos': _cfgr( url='', - mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), test_input_size=(3, 320, 320), crop_pct=0.95), + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), test_input_size=(3, 320, 320), first_conv='stem.conv', crop_pct=0.95), 'regnetz_d8_evob': _cfgr( url='', mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), test_input_size=(3, 320, 320), crop_pct=0.95),