├── .darglint ├── .github └── workflows │ ├── coverage.yml │ ├── release.yml │ ├── test_pypi.yml │ └── tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── LICENSE ├── README.md ├── docs ├── build-api-docs.sh ├── conf.py ├── index.rst └── requirements.txt ├── examples ├── bert.py ├── gpt2.py ├── requirements.txt ├── t5.py ├── utils.py └── vision.py ├── flops_profiler ├── __init__.py ├── __version__.py ├── profiler.py └── requirements.txt ├── poetry.lock ├── pyproject.toml ├── requirements.txt ├── setup.py └── tests ├── __init__.py ├── lenet5.py ├── test_multi_devices.py └── test_profiler.py /.darglint: -------------------------------------------------------------------------------- 1 | [darglint] 2 | strictness = short 3 | -------------------------------------------------------------------------------- /.github/workflows/coverage.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.github/workflows/coverage.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.github/workflows/test_pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.github/workflows/test_pypi.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/README.md -------------------------------------------------------------------------------- /docs/build-api-docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/docs/build-api-docs.sh -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /examples/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/examples/bert.py -------------------------------------------------------------------------------- /examples/gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/examples/gpt2.py -------------------------------------------------------------------------------- /examples/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/examples/requirements.txt -------------------------------------------------------------------------------- /examples/t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/examples/t5.py -------------------------------------------------------------------------------- /examples/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/examples/utils.py -------------------------------------------------------------------------------- /examples/vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/examples/vision.py -------------------------------------------------------------------------------- /flops_profiler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/flops_profiler/__init__.py -------------------------------------------------------------------------------- /flops_profiler/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/flops_profiler/__version__.py -------------------------------------------------------------------------------- /flops_profiler/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/flops_profiler/profiler.py -------------------------------------------------------------------------------- /flops_profiler/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/lenet5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/tests/lenet5.py -------------------------------------------------------------------------------- /tests/test_multi_devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/tests/test_multi_devices.py -------------------------------------------------------------------------------- /tests/test_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cli99/flops-profiler/HEAD/tests/test_profiler.py --------------------------------------------------------------------------------