{"categories":[{"label":"Artificial Intelligence","url":"https://skillfed.io/packages/category/scientific-engineering-artificial-intelligence/4"}],"enrichment":{"capability":"Generates detailed summaries of PyTorch neural network models, showing layer names, input/output shapes, parameter counts, and computational operations\u2014similar to Keras's model.summary() API.","skillfed_tags":["pytorch-debugging","model-inspection","deep-learning-tools"],"use_cases":["Debug neural network architecture by viewing layer-by-layer shapes and parameter counts during model development","Estimate model memory footprint and computational cost before training on expensive hardware","Document model structure for papers or reports by capturing formatted summary output","Verify that model layers are connected correctly and data flows as expected through the network","Compare parameter efficiency across different model designs or architectures"],"what_it_does":"Torchinfo is a PyTorch utility that generates human-readable summaries of neural network models. It displays layer-by-layer information including input and output shapes, parameter counts, and multiply-accumulate operations (Mult-Adds), helping developers understand and debug their model architectures. Unlike the basic print(model) output, torchinfo provides a structured, Keras-like summary view.\n\nThe package accepts either input tensor shapes or actual input data, performs a forward pass through the model to gather statistics, and returns a ModelStatistics object with configurable output columns. It supports RNNs, LSTMs, branching architectures, and nested layers, with options for verbose mode, custom display widths, and Jupyter Notebook integration.","worth_installing":"Yes. Torchinfo is a lightweight, actively maintained debugging tool with no dependencies, permissive licensing, and broad PyTorch version support. Install it if you develop or analyze PyTorch models and want structured visibility into layer structure and computational cost."},"id":"torchinfo","links":{"html":"https://skillfed.io/packages/torchinfo","md":"https://skillfed.io/packages/torchinfo.md","pypi":"https://pypi.org/project/torchinfo/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2023-05-14","license_spdx":null,"license_treatment":"permissive","name":"torchinfo","python_support":"supports_current","summary":"Model summary in PyTorch, based off of the original torchsummary."},"popularity":{"monthly_downloads":653131,"position":5538,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"1.8.0"}