FlopCo is a Python library that aims to make FLOPs and MACs counting simple and accessible for Pytorch neural networks. Moreover FlopCo allows to collect other useful model statistics, such as number of parameters, shapes of layer inputs/outputs, etc.
- numpy
- tensorflow>=2.0
pip install flopco-keras
from flopco_keras import FlopCoKeras
import tensorflow as tf
model = tf.keras.applications.ResNet101()
stats = FlopCoKeras(model)
print(f"FLOPs: {stats.total_flops}")
print(f"MACs: {stats.total_macs}")
print(f"Relative FLOPs: {stats.relative_flops}")List of estimated statistics includes:
- total number of FLOPs/MACs
- number of FLOPs/MACs for each layer
- relative number of FLOPs/MACs for each layer
Make sure your tf.keras model is builded properly
MACS for:
- ResNet50: 3879147569 (3.8B)
- ResNet101: 7601604657 (7.6B)
- ResNet152: 11326470193 (11.3B)
Same as eq here
Project is distributed under MIT License