{"repo":"TylerYep/torchinfo","free":true,"listed":false,"github":"https://github.com/TylerYep/torchinfo","clone":"git clone https://github.com/TylerYep/torchinfo.git","description":"View model summaries in PyTorch!","language":"Python","stars":2947,"topics":["pytorch","torchsummary","torch","keras","visualization","torchvision","torch-summary","torchinfo","python"],"license":"MIT","category":"machine-learning","readme_excerpt":"torchinfo (formerly torch-summary) Torchinfo provides information complementary to what is provided by print(your model) in PyTorch, similar to Tensorflow's model.summary() API to view the visualization of the model, which is helpful while debugging your network. In this project, we implement a similar functionality in PyTorch and create a clean, simple interface to use in your projects. This is a completely rewritten version of the original torchsummary and torchsummaryX projects by @sksq96 and @nmhkahn. This project addresses all of the issues and pull requests left on the original projects by introducing a completely new API. This project currently supports Python 3.12+ and PyTorch versions 2.11+. Older versions support Python 3.8+ and PyTorch versions 1.4.0+, but new changes and bug fixes are no longer backported to these versions. Usage Alternatively, via conda: How To Use Note: if you are using a Jupyter Notebook or Google Colab, summary(model, ...) must be the returned value of the cell. If it is not, you should wrap the summary in a print(), e.g. print(summary(model, ...)) . See tests/jupyter test.ipynb for examples. This version now supports: - RNNs, LSTMs, and other recursive layers - Branching output used to explore model layers using specified depths - Returns ModelStatistics object containing all summary data fields - Configurable rows/columns - Jupyter Notebook / Google Colab Other new features: - Verbose mode to show weights and bias layers - Accepts either inp","default_branch":null,"files":null,"tree":[],"storefront":"/r/TylerYep","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/TylerYep/torchinfo/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}