/Count-FLOPs-Params

This is a pytorch type of counting FLOPs and Params of nets。

Primary LanguagePythonMIT LicenseMIT

Stargazers