TylerYep/torchinfo
A PyTorch utility that renders model summaries showing layer dimensions, parameter counts, and output shapes.

Velocity · 7d
+1.3
★ / day
Trend
→steady
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.