← all repositories

sksq96/pytorch-summary

PyTorch utility that prints model architecture summaries with layer output shapes and parameter counts.

4.1k stars Python ML Frameworks
pytorch-summary
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.

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