search

sovrasov / flops-counter.pytorch

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

Unify flops computation approach #64

Closed sovrasov closed 3 years ago

sovrasov commented 3 years ago

Now all known modules are treated as atomic ones. Resolves #62