facebookarchive / caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.
https://caffe2.ai
Apache License 2.0
8.42k stars 1.94k forks source link

Visualize network graph #2156

Open kampelmuehler opened 6 years ago

kampelmuehler commented 6 years ago

Is there a way to visualize a graph for larger convolutional neural networks? For larger ResNets net_drawer freezes, since I suppose the output files become rather large. Any ideas?

wangzhangup commented 6 years ago

I save the net as .svg format.

Code: graph = net_drawer.GetPydotGraph(net) graph.write_svg('net.svg')

The graph save as net.svg