Contents

torch-summary 1.4.5

0

Model summary in PyTorch, based off of the original torchsummary.

Model summary in PyTorch, based off of the original torchsummary.

Stars: 2552, Watchers: 2552, Forks: 119, Open Issues: 46

The TylerYep/torchinfo repo was created 4 years ago and the last code push was 3 days ago.
The project is very popular with an impressive 2552 github stars!

How to Install torch-summary

You can install torch-summary using pip

pip install torch-summary

or add it to a project with poetry

poetry add torch-summary

Package Details

Author
Tyler Yep @tyleryep
License
MIT
Homepage
https://github.com/tyleryep/torchinfo
PyPi:
https://pypi.org/project/torch-summary/
GitHub Repo:
https://github.com/tyleryep/torchinfo

Classifiers

No  torch-summary  pypi packages just yet.

Errors

A list of common torch-summary errors.

Code Examples

Here are some torch-summary code examples and snippets.

GitHub Issues

The torch-summary package has 46 open issues on GitHub

  • get_total_memory_used fails to handle list of str
  • Support forward with multiple arguments
  • Support CUDA in GitHub Actions testing

See more issues on GitHub

Related Packages & Articles

onnx 1.17.0

Open Neural Network Exchange (ONNX) is an open ecosystem that empowers AI developers to choose the right tools as their project evolves. ONNX provides an open source format for AI models, both deep learning and traditional ML. It defines an extensible computation graph model, as well as definitions of built-in operators and standard data types. Currently we focus on the capabilities needed for inferencing (scoring).

livelossplot 0.5.5

Live training loss plot in Jupyter Notebook for Keras, PyTorch and others.

horovod 0.28.1

Horovod is a powerful distributed training framework for Python that allows you to train deep learning models across multiple GPUs and servers quickly and efficiently. It falls under the category of distributed computing libraries. Built on top of TensorFlow, PyTorch, and other popular deep learning frameworks, Horovod simplifies the process of scaling up your model training by handling the complexities of distributed training under the hood.

barbar 0.2.1

Progress bar for deep learning training iterations