├── .gitignore ├── LICENSE.txt ├── MANIFEST.in ├── README.md ├── flopco ├── __init__.py ├── compute_layer_flops.py ├── examples │ └── model_stats.ipynb └── flopco.py ├── requirements.txt ├── setup.cfg └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /flopco/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/flopco/__init__.py -------------------------------------------------------------------------------- /flopco/compute_layer_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/flopco/compute_layer_flops.py -------------------------------------------------------------------------------- /flopco/examples/model_stats.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/flopco/examples/model_stats.ipynb -------------------------------------------------------------------------------- /flopco/flopco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/flopco/flopco.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [metadata] 2 | description-file = README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliagusak/flopco-pytorch/HEAD/setup.py --------------------------------------------------------------------------------