├── .gitignore ├── LICENSE ├── README.md ├── setup.py └── torchsummary ├── __init__.py ├── tests ├── test_models │ └── test_model.py └── unit_tests │ └── torchsummary_test.py └── torchsummary.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | *.pyc 3 | .vscode/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/setup.py -------------------------------------------------------------------------------- /torchsummary/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/torchsummary/__init__.py -------------------------------------------------------------------------------- /torchsummary/tests/test_models/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/torchsummary/tests/test_models/test_model.py -------------------------------------------------------------------------------- /torchsummary/tests/unit_tests/torchsummary_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/torchsummary/tests/unit_tests/torchsummary_test.py -------------------------------------------------------------------------------- /torchsummary/torchsummary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sksq96/pytorch-summary/HEAD/torchsummary/torchsummary.py --------------------------------------------------------------------------------