/torchinfo

View model summaries in PyTorch!

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.