KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #developer-tools #ai-agent #claude-code #codex #typescript #ai-agents #deepseek-harness #mcp #cli #rust

keras-flops

★ 56 Open GitHub ↗

FLOPs calculator with tf.profiler for neural network architecture written in tensorflow 2.2+ (tf.keras)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# keras-flops

![](https://github.com/tokusumi/keras-flops/workflows/Tests/badge.svg)
[![PyPI version](https://badge.fury.io/py/keras-flops.svg)](https://badge.fury.io/py/keras-flops)

FLOPs calculator for neural network architecture written in tensorflow (tf.keras) v2.2+

This stands on the shoulders of giants, [tf.profiler](https://www.tensorflow.org/api_docs/python/tf/compat/v1/profiler/Profiler).

## Requirements

* Python 3.6+
* Tensorflow 2.2+

## Installation

Using pip:

```
pip install keras-flops
```

## Example

See colab examples [here](https://github.com/tokusumi/keras-flops/tree/master/notebooks) in details.

```python
from tensorflow.keras import Model, Input
from tensorflow.keras.layers import Dense, Flatten, Conv2D, MaxPooling2D, Dropout

from keras_flops import get_flops

# build model
inp = Input((32, 32, 3))
x = Conv2D(32, kernel_size=(3, 3), activation="relu")(inp)
x = Conv2D(64, (3, 3), activation="relu")(x)
x = MaxPooling2D(pool_size=(2, 2))(x)
x = Dropout(0.25)(x)
x = Flatten()(x)
x = Dense(128, activation="relu")(x)
x = Dropout(0.5)(x)
out = Dense(10, activation="softmax")(x)
model = Model(inp, out)

# Calculae FLOPS
flops = get_flops(model, batch_size=1)
print(f"FLOPS: {flops / 10 ** 9:.03} G")
# >>> FLOPS: 0.0338 G
```

## Support

Support `tf.keras.layers` as follows,

## Not supported

Not support `tf.keras.layers` as follows. They are calculated as zero or smaller value than correct value.

keras-flops

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README