Skip to content

Commit

Permalink
Add DeiT results
Browse files Browse the repository at this point in the history
VainF committed Jul 7, 2024
1 parent 41f332f commit 98a23ed
Showing 2 changed files with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -121,6 +121,13 @@ python evaluate.py --model PATH_TO_PRUNED_MODEL --interpolation bicubic

The ConvNext models in our paper were finetuned using the [official implementation](https://github.com/facebookresearch/ConvNeXt). We provide an isolated script ``convnext_train.py`` for ConvNext finetuning and will unify the training scripts in the future.

## Results on DeiT

<div align="center">
<img src="assets/exp_deit.png" width="90%">
</div>


## Acknowledgement

This implementation is based on [Timm](https://github.com/huggingface/pytorch-image-models) and [Torchvision Reference](https://github.com/pytorch/vision/tree/main/references/classification).
Binary file added assets/exp_deit.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 98a23ed

Please sign in to comment.