From b544ad4d3fcd02057ab9f43b118290f2a089566f Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 6 Oct 2021 21:14:59 -0700 Subject: [PATCH] regnetz model default cfg tweaks --- timm/models/byoanet.py | 2 +- timm/models/byobnet.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/timm/models/byoanet.py b/timm/models/byoanet.py index 8c816f6e..8b629dc4 100644 --- a/timm/models/byoanet.py +++ b/timm/models/byoanet.py @@ -69,7 +69,7 @@ default_cfgs = { 'haloregnetz_b': _cfg( url='', - input_size=(3, 224, 224), pool_size=(7, 7), min_input_size=(3, 224, 224), crop_pct=0.94), + first_conv='stem.conv', input_size=(3, 224, 224), pool_size=(7, 7), min_input_size=(3, 224, 224), crop_pct=0.94), 'trionet50ts_256': _cfg( url='', fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index 4ac6ece3..4363709f 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -137,12 +137,12 @@ default_cfgs = { # experimental models, likely to change ot be removed 'regnetz_b': _cfgr( - url='', + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/regnetz_b_raa-677d9606.pth', mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), - input_size=(3, 224, 224), pool_size=(7, 7), first_conv='stem.conv'), + input_size=(3, 224, 224), pool_size=(7, 7), test_input_size=(3, 288, 288), first_conv='stem.conv', crop_pct=0.95), 'regnetz_c': _cfgr( - url='', - imean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), first_conv='stem.conv'), + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/regnetz_c_rab_256-6bdb3c01.pth', + 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_d': _cfgr( url='', mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)),