Machine Learning for Mobile

About bottlenecks

Here, we will try to understand how the retraining process works. The ImageNet models we are using are made up of many layers stacked on top of each other. These layers are pre-trained and already have sufficient information that will help in image classification. All we are trying to do is train the very last layer, final_training_ops , when all the previous layers retrain their already trained state.

The following screenshot is taken from TensorBoard. You can open TensorBoard in your browser to get a better look at it. You will find it in the Graphs tab:

In the preceding diagram, the softmax node on the left-hand side is ...