/torchinfo

View model summaries in PyTorch!

Primary LanguagePythonMIT LicenseMIT

Watchers