search

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework
MIT License
2.82k stars 307 forks source link

Counting bias addition ops for linear layers #60

Closed lzrvch closed 3 years ago

lzrvch commented 3 years ago

Seems like you add bias addition ops to the total FLOPs count for convolutional and RNN modules, but it looks like it is not the case for linear layers. Is that something to be addressed?

sovrasov commented 3 years ago

Hi! That's a bug indeed. Thanks for catching it.