FLOPs counting for the converted model - Githubissues

microsoft / hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.

MIT License

3.32k stars 274 forks source link

FLOPs counting for the converted model #664

Open ChuniHiro opened 1 year ago

ChuniHiro commented 1 year ago

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 commented 1 year ago

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?