/thop

Profile PyTorch models for FLOPs and parameters, helping to evaluate computational efficiency and memory usage.

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

Watchers