Update README.md (#3)
Browse files- Update README.md (4d12ff3f241c17e84bb666c81b29d6ddc5f3e3f2)
README.md
CHANGED
|
@@ -24,6 +24,23 @@ Please observe [original license](https://github.com/apple/ml-fastvit/blob/8af59
|
|
| 24 |
- **Original:** https://github.com/apple/ml-fastvit
|
| 25 |
- **Dataset:** ImageNet-1k
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
## Citation
|
| 28 |
```bibtex
|
| 29 |
@inproceedings{vasufastvit2023,
|
|
|
|
| 24 |
- **Original:** https://github.com/apple/ml-fastvit
|
| 25 |
- **Dataset:** ImageNet-1k
|
| 26 |
|
| 27 |
+
## Evaluation - Variants
|
| 28 |
+
|
| 29 |
+
| Variant | Parameters | Size (MB) | Weight precision | Act. precision | Δ Pytorch acc |
|
| 30 |
+
| ------------------------------------------------------- | ---------: | --------: | ---------------- | -------------- | ------------- |
|
| 31 |
+
| T8 | 3.6M | 7.8 | Float16 | Float16 | -0.9% |
|
| 32 |
+
| MA36 | 42.7M | 84 | Float16 | Float16 | -0.06% |
|
| 33 |
+
|
| 34 |
+
|
| 35 |
+
## Evaluaition - Inference time
|
| 36 |
+
|
| 37 |
+
| Variant | Device | OS | Inference time (ms) | Dominant compute unit |
|
| 38 |
+
| ------- | -------------------- | ---- | ------------------: | --------------------- |
|
| 39 |
+
| T8 | iPhone 12 Pro Max | 17.5 | 0.79 | Neural Engine |
|
| 40 |
+
| T8 | M3 Max | 14.4 | 0.62 | Neural Engine |
|
| 41 |
+
| MA36 | iPhone 12 Pro Max | 18.0 | 4.50 | Neural Engine |
|
| 42 |
+
| MA36 | M3 Max | 15.0 | 2.99 | Neural Engine |
|
| 43 |
+
|
| 44 |
## Citation
|
| 45 |
```bibtex
|
| 46 |
@inproceedings{vasufastvit2023,
|