Plot LSTM network


#1

Does anyone plot LSTM network successfully?

I follow the tutorial to construct LSTM network: https://mxnet.incubator.apache.org/tutorials/scala/char_lstm.html

When I call Visualization.plotNetwork(…).render(…), the program execution does not stop so that I can’t get any plotting.
Does anyone know how to plot the network correctly?


#2

What are you passing to plotNetwork()? My guess is that you’re unrolling the LSTM for a very long sequence.


#3

You are right. Thank you.

I tried to modify the engine parameter and got the plotting successful.
But the plotting is huge and unreadable.
I just want to plot the network roughly like this page: http://dmlc.ml/mxnet/2015/11/15/char-lstm-in-julia.html