Skip to main content
3 of 6
added 286 characters in body
Martin Thoma
  • 19.8k
  • 37
  • 99
  • 171

Tensorflow

If the neural network is given as a Tensorflow graph, then you can visualize this graph with TensorBoard: https://www.tensorflow.org/versions/r0.9/how_tos/graph_viz/index.html

Here is how the MNIST CNN looks like:

enter image description here

You can add names / scopes (like "dropout", "softmax", "fc1", "conv1", "conv2") yourself.

Interpretation

Each box is a layer with parameters that can be learned. For inference, information flows from bottom to the top. Ellipses are layers which do not contain learned parameters.

The color of the boxes does not have a meaning.

I'm not sure of the value of the dashed small boxes ("gradients", "Adam", "save").

Martin Thoma
  • 19.8k
  • 37
  • 99
  • 171