Search code examples
tensorflowdeep-learningpytorchvisualizationcode-visualization

How can I visualize network architectures effectively?


Is there some sort of software that can do so? Specifically, I would like to visualize Resnet18. Is there no other way other than to just draw it myself? Here is an example of what I want to see:

Sample Architecture Visualization

Sample Architecture Visualization


Solution

  • You can use this one : http://alexlenail.me/NN-SVG/LeNet.html . It lets you visualize neural networks by letting you modify several parameters and finally lets you export the architectures as SVG files. You can also choose between 3 visualization styles, namely FCNN, LeNet & AlexNet.