├── .gitignore ├── README.md ├── requirements.txt ├── setup.py └── torchstat2 ├── __init__.py ├── __main__.py ├── compute_flops.py ├── compute_madd.py ├── compute_memory.py ├── model_hook.py ├── reporter.py ├── stat_tree.py ├── statistics.py └── throughput.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/setup.py -------------------------------------------------------------------------------- /torchstat2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/__init__.py -------------------------------------------------------------------------------- /torchstat2/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/__main__.py -------------------------------------------------------------------------------- /torchstat2/compute_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/compute_flops.py -------------------------------------------------------------------------------- /torchstat2/compute_madd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/compute_madd.py -------------------------------------------------------------------------------- /torchstat2/compute_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/compute_memory.py -------------------------------------------------------------------------------- /torchstat2/model_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/model_hook.py -------------------------------------------------------------------------------- /torchstat2/reporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/reporter.py -------------------------------------------------------------------------------- /torchstat2/stat_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/stat_tree.py -------------------------------------------------------------------------------- /torchstat2/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/statistics.py -------------------------------------------------------------------------------- /torchstat2/throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Phoenix8215/torchstat2/HEAD/torchstat2/throughput.py --------------------------------------------------------------------------------