timm
/

Image Classification
timm
PyTorch
Safetensors
rwightman HF staff commited on
Commit
509be83
1 Parent(s): 994c695

Update model config and README

Browse files
Files changed (1) hide show
  1. README.md +12 -10
README.md CHANGED
@@ -17,8 +17,8 @@ A Hiera image classification model. Pretrained on ImageNet-1k with Self-Supervis
17
  - **Model Type:** Image classification / feature backbone
18
  - **Model Stats:**
19
  - Params (M): 51.5
20
- - GMACs: 9.4
21
- - Activations (M): 30.4
22
  - Image size: 224 x 224
23
  - **Dataset:** ImageNet-1k
24
  - **Papers:**
@@ -118,14 +118,16 @@ output = model.forward_head(output, pre_logits=True)
118
  ## Model Comparison
119
  ### By Top-1
120
 
121
- |model |top1 |top1_err|top5 |top5_err|param_count|
122
- |---------------------------------|------|--------|------|--------|-----------|
123
- |hiera_huge_224.mae_in1k_ft_in1k |86.834|13.166 |98.01 |1.99 |672.78 |
124
- |hiera_large_224.mae_in1k_ft_in1k |86.042|13.958 |97.648|2.352 |213.74 |
125
- |hiera_base_plus_224.mae_in1k_ft_in1k|85.134|14.866 |97.158|2.842 |69.9 |
126
- |hiera_base_224.mae_in1k_ft_in1k |84.49 |15.51 |97.032|2.968 |51.52 |
127
- |hiera_small_224.mae_in1k_ft_in1k |83.884|16.116 |96.684|3.316 |35.01 |
128
- |hiera_tiny_224.mae_in1k_ft_in1k |82.786|17.214 |96.204|3.796 |27.91 |
 
 
129
 
130
  ## Citation
131
  ```bibtex
 
17
  - **Model Type:** Image classification / feature backbone
18
  - **Model Stats:**
19
  - Params (M): 51.5
20
+ - GMACs: 8.8
21
+ - Activations (M): 25.1
22
  - Image size: 224 x 224
23
  - **Dataset:** ImageNet-1k
24
  - **Papers:**
 
118
  ## Model Comparison
119
  ### By Top-1
120
 
121
+ |model |top1 |top5 |param_count|
122
+ |---------------------------------|------|------|-----------|
123
+ |hiera_huge_224.mae_in1k_ft_in1k |86.834|98.01 |672.78 |
124
+ |hiera_large_224.mae_in1k_ft_in1k |86.042|97.648|213.74 |
125
+ |hiera_base_plus_224.mae_in1k_ft_in1k|85.134|97.158|69.9 |
126
+ |hiera_small_abswin_256.sbb2_e200_in12k_ft_in1k |84.912|97.260|35.01 |
127
+ |hiera_small_abswin_256.sbb2_pd_e200_in12k_ft_in1k |84.560|97.106|35.01 |
128
+ |hiera_base_224.mae_in1k_ft_in1k |84.49 |97.032|51.52 |
129
+ |hiera_small_224.mae_in1k_ft_in1k |83.884|96.684|35.01 |
130
+ |hiera_tiny_224.mae_in1k_ft_in1k |82.786|96.204|27.91 |
131
 
132
  ## Citation
133
  ```bibtex