/pytorch-OpCounter

Count the FLOPs of your model.

Primary LanguagePythonMIT LicenseMIT

Watchers