← all repositories

TylerYep/torchinfo

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

2.9k stars Python ML FrameworksLLMOps · Eval
torchinfo
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.