Skip to content
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.

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

FLOPs counting for the converted model #664

Open
ChuniHiro opened this issue Dec 16, 2022 · 1 comment
Open

FLOPs counting for the converted model #664

ChuniHiro opened this issue Dec 16, 2022 · 1 comment

Comments

@ChuniHiro
Copy link

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!

@interesaaat
Copy link
Collaborator

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
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants