README.md

---
license: apache-2.0
datasets:
- imagenet-1k
metrics:
- accuracy
pipeline_tag: image-classification
tags:
- pytorch
- torch-dag
---
# Model Card for convnetxtv2_nano_pruned_19

This is a prunned version of the [timm/convnextv2_nano.fcmae_ft_in22k_in1k](https://huggingface.co/timm/convnextv2_nano.fcmae_ft_in22k_in1k) model in a [toch-dag](https://github.com/TCLResearchEurope/torch-dag) format.

This model has rougly 19% of the original model FLOPs with minimal metrics drop.


| Model                                    | KMAPPs*         | M Parameters   | Accuracy (224x224)      |
| -----------                              | -----------     | -----------    | ------------------      |
| **timm/convnextv2_nano.fcmae_ft_in22k_in1 (baseline)**   | 97.9           | 15.6           | 81.97%                  |
| **convnetxtv2_nano_pruned_19 (ours)**          | 18.6 **(19%)** | 3.8 **(24%)** | 75.31% **(↓ 6.66%)**    |


\***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/convnextv2_nano.fcmae_ft_in22k_in1k](https://huggingface.co/timm/convnextv2_nano.fcmae_ft_in22k_in1k)

### Model Sources
- **Repository:** [timm/convnextv2_nano.fcmae_ft_in22k_in1k](https://huggingface.co/timm/convnextv2_nano.fcmae_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/convnetxtv2_nano_pruned_19
```

and now You are ready to load the model:

```
import torch_dag
import torch

model = torch_dag.io.load_dag_from_path('./convnetxtv2_nano_pruned_19')

model.eval()
out = model(torch.ones(1, 3, 224, 224))
print(out.shape)
```