├── .gitignore ├── LICENSE ├── README.md ├── examples ├── CNN.py └── Transformer.py ├── pytorch_model_summary ├── __init__.py ├── hierarchical_summary.py └── model_summary.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | build/ 3 | dist/ 4 | *.egg-info/ 5 | __pycache__/ 6 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/README.md -------------------------------------------------------------------------------- /examples/CNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/examples/CNN.py -------------------------------------------------------------------------------- /examples/Transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/examples/Transformer.py -------------------------------------------------------------------------------- /pytorch_model_summary/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/pytorch_model_summary/__init__.py -------------------------------------------------------------------------------- /pytorch_model_summary/hierarchical_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/pytorch_model_summary/hierarchical_summary.py -------------------------------------------------------------------------------- /pytorch_model_summary/model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/pytorch_model_summary/model_summary.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amarczew/pytorch_model_summary/HEAD/setup.py --------------------------------------------------------------------------------