search

fossasia / visdom

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.
Apache License 2.0
9.98k stars 1.14k forks source link

[Feature Request] Neural network structure visualization   #510

Open shayxurui opened 5 years ago

shayxurui commented 5 years ago

Will support the visualization of neural network structure?

JackUrb commented 5 years ago

We would love to support architecture visualization, but we don't have any of this functionality at the moment.

shayxurui commented 5 years ago

thanks. Expect to support network structure visualization

Kylin9511 commented 5 years ago

Actually, you can use Hidden Layer or other tools to generate an image of your network structure, and use visdom to load it.

JackUrb commented 5 years ago

@luzhilin19951120 - what is Hidden Layer? I can't seem to find it in a search. I'm trying to evaluate existing workflows for model evaluation before we start to implement a native solution, so if you could link it that would be amazing.

dzimmerer commented 5 years ago

If you are using pytorch, we are (currently and for some while now) working on a logging framework/tool which heavily relies on and complements pytorch & visdom, and have also already implemented a way to log your network structure:

https://trixi.readthedocs.io/en/latest/_api/trixi.logger.html#trixi.logger.visdom.pytorchvisdomlogger.PytorchVisdomLogger.plot_model_structure

(and some brazen advertisement: Feel free to check out the whole "framework": https://mic-dkfz.github.io/trixi/ :wink: )

Kylin9511 commented 5 years ago

@JackUrb Actually HiddenLayer is a NN structure visualization framework:

https://github.com/waleedka/hiddenlayer

But I think the link proposed by @dzimmerer seems to be more fancy~ I would give both of them a try~

dizcza commented 3 years ago

@dzimmerer seems like your loggers have no longer plot_model_structure function, right?