PyTorch Model Summary
While I was debugging a complex CNN architecture, and needed to quickly check the number of parameters and layer shapes. The issue was, PyTorch doesn’t have a built-in summary function like Keras does. After experimenting with different solutions, I found several effective ways to visualize model architecture. In this article, I’ll share five practical methods … Read more >>