Fixed eca_botnext26ts_256 weights added, 79.27

pull/933/head
Ross Wightman 3 years ago
parent c02334d9fa
commit b6caa356d2

@ -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(

Loading…
Cancel
Save