From b6caa356d2dcac2c02bc1c81b9d3ffbae3fc50ad Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 19 Oct 2021 12:44:28 -0700 Subject: [PATCH] Fixed eca_botnext26ts_256 weights added, 79.27 --- timm/models/byoanet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/byoanet.py b/timm/models/byoanet.py index c7a5c53e..dfcba46f 100644 --- a/timm/models/byoanet.py +++ b/timm/models/byoanet.py @@ -40,7 +40,7 @@ default_cfgs = { url='', fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), 'eca_botnext26ts_256': _cfg( - url='', + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/eca_botnext26ts_c_256-95a898f6.pth', fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), 'halonet_h1': _cfg(url='', input_size=(3, 256, 256), pool_size=(8, 8), min_input_size=(3, 256, 256)), @@ -122,7 +122,7 @@ model_cfgs = dict( act_layer='silu', attn_layer='eca', self_attn_layer='bottleneck', - self_attn_kwargs=dict() + self_attn_kwargs=dict(dim_head=16) ), halonet_h1=ByoModelCfg(