Update model config and README
Browse files
README.md
CHANGED
@@ -17,8 +17,8 @@ A Hiera image feature model. Pretrained on ImageNet-1k with Self-Supervised Mask
|
|
17 |
- **Model Type:** Image classification / feature backbone
|
18 |
- **Model Stats:**
|
19 |
- Params (M): 50.8
|
20 |
-
- GMACs:
|
21 |
-
- Activations (M):
|
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 |
|
122 |
-
|
123 |
-
|hiera_huge_224.mae_in1k_ft_in1k |86.834|
|
124 |
-
|hiera_large_224.mae_in1k_ft_in1k |86.042|
|
125 |
-
|hiera_base_plus_224.mae_in1k_ft_in1k|85.134|
|
126 |
-
|
|
127 |
-
|
|
128 |
-
|
|
|
|
|
|
129 |
|
130 |
## Citation
|
131 |
```bibtex
|
|
|
17 |
- **Model Type:** Image classification / feature backbone
|
18 |
- **Model Stats:**
|
19 |
- Params (M): 50.8
|
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
|