GitHub repository: luke396/torchinfo
Primary language: Python
Description: View model summaries in PyTorch!