Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
YifanXu74 authored Sep 14, 2021
1 parent 7fba51b commit a676ec1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
This repository contains the PyTorch code for Evo-ViT.

This work proposes a slow-fast token evolution approach to accelerate vanilla vision transformers of both flat and deep-narrow structures without additional pre-training and fine-tuning procedures. For details please see [Evo-ViT: Slow-Fast Token Evolution for Dynamic Vision Transformer](https://arxiv.org/abs/2108.01390) by Yifan Xu*, Zhijie Zhang*, Mengdan Zhang, Kekai Sheng, Ke Li, Weiming Dong, Liqing Zhang, Changsheng Xu, and Xing Sun.
![intro](/imgs/method.png)
![intro](imgs/method.png)

Our code is based on [pytorch-image-models](https://github.com/rwightman/pytorch-image-models), [DeiT](https://github.com/facebookresearch/deit), and [LeViT](https://github.com/facebookresearch/LeViT).

Expand Down Expand Up @@ -124,7 +124,7 @@ Add parameter '--layer-wise-prune' if the visualized model is not trained with l

The visualization results of Evo-ViT-S are as follows:

![result](/imgs/results.png)
![result](imgs/results.png)


# Citation
Expand Down

0 comments on commit a676ec1

Please sign in to comment.