sksq96/pytorch-summary
PyTorch utility that prints model architecture summaries with layer output shapes and parameter counts.

Velocity · 7d
+1.4
★ / day
Trend
→steady
star history
This library replicates Keras model.summary() functionality for PyTorch. It performs a forward pass through the model to display each layer’s output dimensions and count trainable parameters, helping developers visualize and debug neural network architectures. The library requires specifying input size to execute the forward pass and generates a tabular view of the model’s structure.