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): 27.1
|
| 20 |
-
- GMACs: 4.
|
| 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): 27.1
|
| 20 |
+
- GMACs: 4.7
|
| 21 |
+
- Activations (M): 14.6
|
| 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
|