/ptflops

Flops counter for convolutional networks in pytorch framework

Primary LanguagePythonMIT LicenseMIT

Stargazers