Flops after discarding tokens

raoyongming / DynamicViT

[NeurIPS 2021] [T-PAMI] DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsification

https://dynamicvit.ivg-research.xyz/

MIT License

551 stars 69 forks source link

Closed King4819 closed 9 months ago

King4819 commented 9 months ago

Hi, I want know how I can get the information of model's flop after discarding tokens. Thanks!

raoyongming commented 9 months ago

The line of code should print the FLOPs after sparsification.