# TylerYep/torchinfo

View model summaries in PyTorch!

Repository: https://github.com/TylerYep/torchinfo
Canonical: https://ross.abutalabs.com/products/torchinfo
Language: Python
License: MIT
License Family: permissive
Topics: pytorch, torchsummary, torch, keras, visualization, torchvision, torch-summary, torchinfo, python
Last push: 2026-08-26T05:52:19+00:00

## Health v2 (maintenance only)
Score: 67/100 (v2, computed 2026-09-03T02:20:16.233290+00:00)
- activity 99, release rhythm 8, longevity 100
- inputs: {"age_days": 2361, "days_push": 7, "days_rel": null, "gap_med": null, "n_releases_24m": 0}
- flags: none
- formula: round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)

## Adoption (not part of the score)
Stars 2947, forks 138 (observed 2026-08-28T04:07:31.797030+00:00)

## What it is
torchinfo is a Python library that provides Keras-style model summaries for PyTorch, showing layer types, input/output shapes, parameter counts, and multiply-add operations. It is a rewritten successor to torchsummary with a cleaner API for inspecting and debugging neural network architectures.

## Use cases
- view pytorch model summary like keras
- count parameters in a pytorch model
- inspect layer shapes of a neural network
- debug pytorch network architecture
- estimate model compute with mult-adds
- replace torchsummary with maintained alternative

## When to choose
- you need a detailed, maintained model summary for PyTorch models
- you want parameter counts and FLOPs per layer for debugging
- you miss TensorFlow's model.summary() when working in PyTorch

## When to avoid
- you use TensorFlow/Keras, which has model.summary() built in
- you need full computational graph visualization rather than a tabular summary
- you require support for very old Python or PyTorch versions no longer backported

## Facets
- artifact type: library
- maturity: active
- function: data-visualization, developer-tools, machine-learning
- domain: deep-learning, machine-learning, developer-tools
- platform: python
- tags: pytorch, model-summary, torchsummary, debugging, neural-networks

## Member repositories
- TylerYep/torchinfo (main) score 67

## Provenance
- Observed fields: from GitHub, fetched 2026-08-28T04:07:31.797030+00:00.
- Health v2: computed from the inputs above; adoption is never an input.
- Inferred fields (summary, facets, guidance): AI-extracted, prompt v1, taxonomy v1, on 2026-08-30T07:32:45.923898+00:00, confidence not recorded.
  - readme: https://github.com/TylerYep/torchinfo (fetched 2026-08-28T04:07:31.797030+00:00, sha 5f67e4872808)
  - registry_pypi: https://pypi.org/pypi/torchinfo/json (fetched 2026-08-29T09:47:59.934797+00:00, sha 095391b5cc77)
- Data as of 2026-08-30T08:39:29.467469+00:00.