View model summaries in PyTorch! - View it on GitHub