/torchprofile

A general and accurate MACs / FLOPs profiler for PyTorch models

Primary LanguagePythonMIT LicenseMIT

Watchers