the flops computed by this code don't match that in the paper

microsoft / CvT

This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.

MIT License

533 stars 120 forks source link

the flops computed by this code don't match that in the paper #8

Open cuixing61 opened 2 years ago

cuixing61 commented 2 years ago

2021-07-20 16:52:11,694:[P:33903]:Rank[0/4] == get_model_complexity_info by ptflops == 2021-07-20 16:52:11,796:[P:33903]:Rank[0/4] => FLOPs: 4.06 GMac, params: 20.0 M
2021-07-20 16:52:11,796:[P:33903]:Rank[0/4] == get_model_complexity_info by ptflops ==

while the paper shows the flops of CvT-13 is 4.5G