search

sovrasov / flops-counter.pytorch

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

Aten backend #133

Closed sovrasov closed 6 months ago

sovrasov commented 9 months ago

This PR adds flops estimation based on aten ops, which makes ptflops indifferent to model architecture.