whai362 / PVT

Official implementation of PVT series
Apache License 2.0
1.73k stars 246 forks source link

Flops Calculation for PVT #28

Closed xwjabc closed 3 years ago

xwjabc commented 3 years ago

Hi, I found in https://github.com/whai362/PVT/issues/1, an example is provided to show the complexity of ViT models (using get_vit_flops()). However, since PVT has multiple scales, I wonder if there is a tool provided to measure the flops for PVT? Thanks!


It seems a duplicate post is created. I will close this one.