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