├── .gitignore ├── README.md ├── example.py ├── model_hook.py ├── model_summary.py ├── models ├── __init__.py └── mobilenet_v2.py ├── module_madd.py └── summary_tree.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/README.md -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/example.py -------------------------------------------------------------------------------- /model_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/model_hook.py -------------------------------------------------------------------------------- /model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/model_summary.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/models/mobilenet_v2.py -------------------------------------------------------------------------------- /module_madd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/module_madd.py -------------------------------------------------------------------------------- /summary_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ceykmc/pytorch_model_summary/HEAD/summary_tree.py --------------------------------------------------------------------------------