mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fixed eca_botnext26ts_256 weights added, 79.27

Browse Source
This commit is contained in:
Ross Wightman 2021-10-19 12:44:28 -07:00
parent c02334d9fa
commit b6caa356d2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/byoanet.py
View File

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