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(