/pytorch-OpCounter

Count the FLOPs of your PyTorch model.

Primary LanguagePythonMIT LicenseMIT

Watchers