TylerYep/torchinfo
A PyTorch utility that renders model summaries showing layer dimensions, parameter counts, and output shapes.
Not currently ranked — collecting fresh signals.
star history
Torchinfo generates detailed summaries of PyTorch models, displaying layer-by-layer information including input/output tensor shapes, parameter counts, and trainable/non-trainable breakdowns. It replicates the functionality of TensorFlow’s model.summary() for PyTorch users. The tool is installed via pip or conda and integrates directly with existing PyTorch model objects for quick inspection during development and debugging.
Frequently asked
- What is TylerYep/torchinfo?
- A PyTorch utility that renders model summaries showing layer dimensions, parameter counts, and output shapes.
- Is torchinfo open source?
- Yes — TylerYep/torchinfo is open source, released under the MIT license.
- What language is torchinfo written in?
- TylerYep/torchinfo is primarily written in Python.
- How popular is torchinfo?
- TylerYep/torchinfo has 2.9k stars on GitHub.
- Where can I find torchinfo?
- TylerYep/torchinfo is on GitHub at https://github.com/TylerYep/torchinfo.