flops-counter.pytorch icon indicating copy to clipboard operation
flops-counter.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon sovrasov

Regarding VIT support

Open ghjeong12 opened this issue 2 years ago • 0 comments

Hi,

Thanks for open sourcing a useful tool! I have been using this for double-checking # of MACs and it worked well for alexnet/vgg/resnet/mobilenet/efficientnet and even convnext.

Now I would like to see whether I can use this for vit since it is supported from torchvision models (https://pytorch.org/vision/0.12/models.html), but I got some errors when I test vit_b_16 using the sample/classficiation.py -- I think in the forward hook.

Could you please check whether it's running well from your side? If not, do you have any plan to support this as well?

ghjeong12 avatar Jun 30 '22 20:06 ghjeong12