├── .gitignore ├── .travis.yml ├── LICENSE ├── README.md ├── detail.md ├── example.py ├── requirements.txt ├── setup.cfg ├── setup.py ├── test_requirements.txt └── torchstat ├── __init__.py ├── __main__.py ├── compute_flops.py ├── compute_madd.py ├── compute_memory.py ├── model_hook.py ├── reporter.py ├── stat_tree.py └── statistics.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/README.md -------------------------------------------------------------------------------- /detail.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/detail.md -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/example.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | pandas 4 | -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/setup.py -------------------------------------------------------------------------------- /test_requirements.txt: -------------------------------------------------------------------------------- 1 | pydocstyle 2 | -------------------------------------------------------------------------------- /torchstat/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/__init__.py -------------------------------------------------------------------------------- /torchstat/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/__main__.py -------------------------------------------------------------------------------- /torchstat/compute_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/compute_flops.py -------------------------------------------------------------------------------- /torchstat/compute_madd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/compute_madd.py -------------------------------------------------------------------------------- /torchstat/compute_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/compute_memory.py -------------------------------------------------------------------------------- /torchstat/model_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/model_hook.py -------------------------------------------------------------------------------- /torchstat/reporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/reporter.py -------------------------------------------------------------------------------- /torchstat/stat_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/stat_tree.py -------------------------------------------------------------------------------- /torchstat/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Swall0w/torchstat/HEAD/torchstat/statistics.py --------------------------------------------------------------------------------