mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Remaining Hiera sbb weights uploaded

sbb2_vit_hiera_weights
Ross Wightman 2024-08-21 10:06:27 -07:00
parent a256e50457
commit dc94cca0e5
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
timm/models/hiera.py
View File

@ -881,11 +881,24 @@ default_cfgs = generate_default_cfgs({
num_classes=0, num_classes=0,
), ),
"hiera_small_abswin_256.sbb2_e200_in12k_ft_in1k": _cfg(
hf_hub_id='timm/',
input_size=(3, 256, 256), crop_pct=0.95,
),
"hiera_small_abswin_256.sbb2_pd_e200_in12k_ft_in1k": _cfg(
hf_hub_id='timm/',
input_size=(3, 256, 256), crop_pct=0.95,
),
"hiera_small_abswin_256.sbb2_e200_in12k": _cfg( "hiera_small_abswin_256.sbb2_e200_in12k": _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
num_classes=11821, num_classes=11821,
input_size=(3, 256, 256), crop_pct=0.95, input_size=(3, 256, 256), crop_pct=0.95,
), ),
"hiera_small_abswin_256.sbb2_pd_e200_in12k": _cfg(
hf_hub_id='timm/',
num_classes=11821,
input_size=(3, 256, 256), crop_pct=0.95,
),
"hiera_base_abswin_256.untrained": _cfg( "hiera_base_abswin_256.untrained": _cfg(
# hf_hub_id='timm/', # hf_hub_id='timm/',
input_size=(3, 256, 256), crop_pct=0.95, input_size=(3, 256, 256), crop_pct=0.95,