Update README.md

README.md

@@ -1,3 +1,72 @@
 ---
 license: apache-2.0
+datasets:
+- imagenet-1k
+metrics:
+- accuracy
+pipeline_tag: image-classification
+tags:
+- pytorch
+- torch-dag
 ---
+# Model Card for efficientformerv2_s2_pruned_72
+
+This is a prunned version of the [timm/efficientformerv2_s2.snap_dist_in1k](https://huggingface.co/timm/efficientformerv2_s2.snap_dist_in1k) model in a [toch-dag](https://github.com/TCLResearchEurope/torch-dag) format.
+
+This model has rougly 72% of the original model FLOPs with minimal metrics drop.
+
+
+| Model                                    | KMAPPs*         | M Parameters   | Accuracy (224x224)      |
+| -----------                              | -----------     | -----------    | ------------------      |
+| **timm/efficientformerv2_s2.snap_dist_in1 (baseline)**   | 50.7           | 12.7           | 82.2%                  |
+| **efficientformerv2_s2_pruned_72 (ours)**          | 36.7 **(72%)** | 10.1 **(80%)** | 81.58% **(↓ 0.62%)**    |
+
+
+\***KMAPPs** thousands of FLOPs per input pixel
+
+`KMAPPs(model) = FLOPs(model) / (H * W * 1000)`, where `(H, W)` is the input resolution.
+
+The accuracy was calculated on the ImageNet-1k validation dataset. For details about image pre-processing, please refer to the original repository.
+## Model Details
+
+### Model Description
+
+
+- **Developed by:** [TCL Research Europe](https://github.com/TCLResearchEurope/)
+- **Model type:** Classification / feature backbone
+- **License:** Apache 2.0
+- **Finetuned from model:** [timm/efficientformerv2_s2.snap_dist_in1k](https://huggingface.co/timm/efficientformerv2_s2.snap_dist_in1k)
+
+### Model Sources
+- **Repository:** [timm/efficientformerv2_s2.snap_dist_in1k](https://huggingface.co/timm/efficientformerv2_s2.snap_dist_in1k)
+
+
+
+## How to Get Started with the Model
+
+To load the model, You have to install [torch-dag](https://github.com/TCLResearchEurope/torch-dag#3-installation) library, which can be done using `pip` by
+
+```
+pip install torch-dag
+```
+
+then, clone this repository
+
+```
+# Make sure you have git-lfs installed (https://git-lfs.com)
+git lfs install
+git clone https://huggingface.co/TCLResearchEurope/efficientformerv2_s2_pruned_72
+```
+
+and now You are ready to load the model:
+
+```
+import torch_dag
+import torch
+
+model = torch_dag.io.load_dag_from_path('./efficientformerv2_s2_pruned_72')
+
+model.eval()
+out = model(torch.ones(1, 3, 224, 224))
+print(out.shape)
+```