/pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Primary LanguagePythonMIT LicenseMIT

Stargazers