search

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework
MIT License
2.83k stars 306 forks source link

Aten backend #133

Closed sovrasov closed 7 months ago

sovrasov commented 10 months ago

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