Update model config and README

README.md

@@ -9,11 +9,10 @@ datasets:
 ---
 # Model card for mobilenetv3_large_100.ra4_e3600_r224_in1k
 
-A MobileNet-V4 image classification model. Trained on ImageNet-1k by Ross Wightman.
+A MobileNet-V3 image classification model. Trained on ImageNet-1k by Ross Wightman.
 
 Trained with `timm` scripts using hyper-parameters inspired by the MobileNet-V4 paper with `timm` enhancements.
 
-NOTE: So far, these are the only known MNV4 weights. Official weights for Tensorflow models are unreleased.
 
 
 ## Model Details
@@ -24,10 +23,10 @@ NOTE: So far, these are the only known MNV4 weights. Official weights for Tensor
   - Activations (M): 4.4
   - Image size: train = 224 x 224, test = 256 x 256
 - **Dataset:** ImageNet-1k
-- **Original:** https://github.com/tensorflow/models/tree/master/official/vision
 - **Papers:**
-  - MobileNetV4 -- Universal Models for the Mobile Ecosystem: https://arxiv.org/abs/2404.10518
   - PyTorch Image Models: https://github.com/huggingface/pytorch-image-models
+  - Searching for MobileNetV3: https://arxiv.org/abs/1905.02244
+  - MobileNetV4 -- Universal Models for the Mobile Ecosystem: https://arxiv.org/abs/2404.10518
 
 ## Model Usage
 ### Image Classification
@@ -177,14 +176,6 @@ output = model.forward_head(output, pre_logits=True)
 
 ## Citation
 ```bibtex
-@article{qin2024mobilenetv4,
-  title={MobileNetV4-Universal Models for the Mobile Ecosystem},
-  author={Qin, Danfeng and Leichner, Chas and Delakis, Manolis and Fornoni, Marco and Luo, Shixin and Yang, Fan and Wang, Weijun and Banbury, Colby and Ye, Chengxi and Akin, Berkin and others},
-  journal={arXiv preprint arXiv:2404.10518},
-  year={2024}
-}
-```
-```bibtex
 @misc{rw2019timm,
   author = {Ross Wightman},
   title = {PyTorch Image Models},
@@ -195,3 +186,20 @@ output = model.forward_head(output, pre_logits=True)
   howpublished = {\url{https://github.com/huggingface/pytorch-image-models}}
 }
 ```
+```bibtex
+@inproceedings{howard2019searching,
+  title={Searching for mobilenetv3},
+  author={Howard, Andrew and Sandler, Mark and Chu, Grace and Chen, Liang-Chieh and Chen, Bo and Tan, Mingxing and Wang, Weijun and Zhu, Yukun and Pang, Ruoming and Vasudevan, Vijay and others},
+  booktitle={Proceedings of the IEEE/CVF international conference on computer vision},
+  pages={1314--1324},
+  year={2019}
+}
+```
+```bibtex
+@article{qin2024mobilenetv4,
+  title={MobileNetV4-Universal Models for the Mobile Ecosystem},
+  author={Qin, Danfeng and Leichner, Chas and Delakis, Manolis and Fornoni, Marco and Luo, Shixin and Yang, Fan and Wang, Weijun and Banbury, Colby and Ye, Chengxi and Akin, Berkin and others},
+  journal={arXiv preprint arXiv:2404.10518},
+  year={2024}
+}
+```