torch-summary 1.4.5
0
Model summary in PyTorch, based off of the original torchsummary.
Contents
Model summary in PyTorch, based off of the original torchsummary.
Stars: 2552, Watchers: 2552, Forks: 119, Open Issues: 46The 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
Related Packages
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