New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

FLOPs counting for the converted model #664

Open

ChuniHiro opened this issue Dec 16, 2022 · 1 comment

ChuniHiro commented Dec 16, 2022

Could you please share some suggestions on FLOPs counting for the converted model?

I have tried:
thop : https://github.com/Lyken17/pytorch-OpCounter
flop-counter : https://github.com/sovrasov/flops-counter.pytorch
pthflops: https://github.com/1adrianb/pytorch-estimate-flops
torchprofile: https://github.com/zhijian-liu/torchprofile
deepspeed: https://www.deepspeed.ai/tutorials/flops-profiler/

Seems non of them support the calculations of the converted models, any advice will be highly appreciated, thanks!

Collaborator

interesaaat commented Dec 19, 2022

Hi! I am not really sure about this, I think that any tool working with pytorch should work. What are the errors you are getting?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment