Skip to content

Commit

Permalink
Upload to PyPi (#10)
Browse files Browse the repository at this point in the history
  • Loading branch information
tokusumi authored Aug 16, 2020
1 parent bb85ceb commit 8be3d27
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 4 deletions.
7 changes: 5 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# 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+

Expand All @@ -13,9 +14,11 @@ This stands on the shoulders of giants, [tf.profiler](https://www.tensorflow.org

## Installation

This implementation is simple thanks to stands on the shoulders of giants, [tf.profiler](https://www.tensorflow.org/api_docs/python/tf/compat/v1/profiler/Profiler). Only one function is defined.
Using pip:

Copy and paste [it](https://github.com/tokusumi/keras-flops/blob/master/keras_flops/flops_calculation.py).
```
pip install keras-flops
```

## Example

Expand Down
13 changes: 11 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,14 +1,23 @@
[tool.poetry]
name = "keras-flops"
version = "0.1.0"
description = ""
version = "0.1.1"
description = "FLOPs calculator for neural network architecture written in tensorflow 2.x (tf.keras)"
authors = ["tokusumi <tksmtoms@gmail.com>"]
license = "MIT"
readme = "README.md"
repository = "https://github.com/tokusumi/keras-flops"
include = [
"LICENSE",
]
keywords = ["tensorflow2", "flops", "profiler"]
classifiers = [
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]

[tool.poetry.dependencies]
python = "^3.6"
Expand Down

0 comments on commit 8be3d27

Please sign in to comment.