Model summary in PyTorch, based off of the original torchsummary.
No known vulnerabilities in the latest version
Based on latest version 1.8.0. If you're running an older version, check OSV for your specific version.
[](/packages/torchinfo)
<a href="/packages/torchinfo"><img src="/api/badges/torchinfo?period=month" alt="PyPI Stats"></a>