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='', url='',
fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)), fixed_input_size=True, input_size=(3, 256, 256), pool_size=(8, 8)),
'eca_botnext26ts_256': _cfg( '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)), 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)), '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', act_layer='silu',
attn_layer='eca', attn_layer='eca',
self_attn_layer='bottleneck', self_attn_layer='bottleneck',
self_attn_kwargs=dict() self_attn_kwargs=dict(dim_head=16)
), ),
halonet_h1=ByoModelCfg( halonet_h1=ByoModelCfg(

Loading…
Cancel
Save