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 beit_base_patch16_224_pruned_65
+
+This is a prunned version of the [timm/beit_base_patch16_224.in22k_ft_in22k_in1k](https://huggingface.co/timm/beit_base_patch16_224.in22k_ft_in22k_in1k) model in a [toch-dag](https://github.com/TCLResearchEurope/torch-dag) format.
+
+This model has rougly 65% of the original model FLOPs with minimal metrics drop.
+
+
+| Model                                    | KMAPPs*         | M Parameters   | Accuracy (224x224)      |
+| -----------                              | -----------     | -----------    | ------------------      |
+| **timm/beit_base_patch16_224.in22k_ft_in22k_in1 (baseline)**   | 673.2           | 86.5           | 85.23%                  |
+| **beit_base_patch16_224_pruned_65 (ours)**          | 438 **(65%)** | 56.7 **(66%)** | 84.53% **(↓ 0.7%)**    |
+
+
+\***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/beit_base_patch16_224.in22k_ft_in22k_in1k](https://huggingface.co/timm/beit_base_patch16_224.in22k_ft_in22k_in1k)
+
+### Model Sources
+- **Repository:** [timm/beit_base_patch16_224.in22k_ft_in22k_in1k](https://huggingface.co/timm/beit_base_patch16_224.in22k_ft_in22k_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/beit_base_patch16_224_pruned_65
+```
+
+and now You are ready to load the model:
+
+```
+import torch_dag
+import torch
+
+model = torch_dag.io.load_dag_from_path('./beit_base_patch16_224_pruned_65')
+
+model.eval()
+out = model(torch.ones(1, 3, 224, 224))
+print(out.shape)
+```