[PyTorch] Using "torchsummary" to plot your model structure
When we using the famous Python framework: PyTorch, to build our model, if we can visualize our model, that's a cool idea.
In this way, we can check our model layer, output shape, and avoid our model mismatch.
Read More »[PyTorch] Using "torchsummary" to plot your model structure