A Python to count flops of pytorch models.
Project description
Flopper - A FLOP counter for PyTorch
An enhanced FLOP counter based on fvcore with additional support to custom modules and simplified API This tool is a lightweight wrapper around fvcore flop counter, which is a PyTorch based FLOP counter. It supports counting FLOPs for custom modules and provides a simplified API.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
flopper-0.1.0.tar.gz
(9.7 kB
view hashes)