Visualizing graphs in TensorBoard

After training, we can visualize our computational graph in TensorBoard, as shown in the following diagram. As you can see, our Model takes input, weights, and biases as input and returns the output. We compute Loss and Accuracy based on the output of the model. We minimize the loss by calculating gradients and updating weights. We can observe all of this in the following diagram:

If we double-click and expand the Model, we can see that we have three hidden layers and one output layer:

Similarly, we can double-click ...

Get Hands-On Deep Learning Algorithms with Python now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.