NTU-P04922004/torch_summary.md

Created November 30, 2019 12:04

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/NTU-P04922004/8d8536f3860923496e461fb4d3fa3515.js"></script>
Save NTU-P04922004/8d8536f3860923496e461fb4d3fa3515 to your computer and use it in GitHub Desktop.

Download ZIP

Model summary in PyTorch

Raw

torch_summary.md

pip install torchsummary or git clone https://github.com/sksq96/pytorch-summary

from torchsummary import summary
summary(your_model, input_size=(channels, H, W))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment