1adrianb / pytorch-estimate-flops

Estimate/count FLOPS for a given neural network using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
305 stars 22 forks source link

Does the project compute the FLOPs or MACs? #15

Open haoyu0408 opened 2 years ago

haoyu0408 commented 2 years ago

I find the project ignores the add operations in matrix product. So does it compute MACs instead of FLOPs?

dani-capellan commented 2 years ago

Comparing results with pytorch-OpCounter implementation, this one seems to be computing MACs instead of FLOPs... Can anyone double check this?