sksq96/pytorch-summary
Model summary in PyTorch similar to `model.summary()` in Keras
This tool helps machine learning engineers and researchers quickly understand the structure and resource requirements of their PyTorch neural network models. You provide your PyTorch model and an example input size, and it outputs a detailed table showing each layer, its output shape, and the number of parameters it contains. This makes debugging and optimizing network architectures much easier.
4,063 stars. No commits in the last 6 months.
Use this if you are a machine learning practitioner working with PyTorch and need a clear, Keras-style summary to visualize your model's layers and parameter counts.
Not ideal if you are not using PyTorch or if you need a more advanced model analysis tool like 'torchinfo'.
Stars
4,063
Forks
411
Language
Python
License
MIT
Category
Last pushed
Mar 02, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/sksq96/pytorch-summary"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
dataflowr/notebooks
code for deep learning courses
jeffheaton/app_deep_learning
T81-558: PyTorch - Applications of Deep Neural Networks @Washington University in St. Louis
dvgodoy/PyTorchStepByStep
Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"
xl0/lovely-tensors
Tensors, for human consumption
rentruewang/koila
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.