IBM Developer Day | Bengaluru | March 14th Register now
Code Pattern
Get the code
By Animesh Singh, Tommy Li, Andrew Butler | Published July 13, 2018
Artificial IntelligenceDeep Learning
Learn how to train your Fashion MNIST model with IBM Fabric for Deep Learning (FfDL) on Kubernetes Cluster configured with GPU. This model is defined in Keras and trained using TensorFlow with FfDL, which is deployed on Kubernetes GPU cluster. FfDL provides a scalable deep learning platform on Kubernetes by giving the option to leverage deep learning libraries like Caffe, Torch, and TensorFlow. In addition, we will show to wrap and deploy that model using Seldon, and then write an app to consume it.
Fashion MNIST is a direct drop-in replacement for the original MNIST dataset. The dataset is made up of 60,000 training examples and 10,000 testing examples, where each example is a 28×28 grayscaled picture of various articles of clothing. The Fashion MNIST dataset is more difficult than the original MNIST, and thus serves as a more complete benchmarking tool.
The model being trained is a CNN with three convolutional layers followed by two dense layers. The job will run for 30 epochs, with a batch size of 128. Once wrapped, the model is ready to take input and return predictions of confidence in the picture being of a certain class.
The web app is designed to take uploaded images, display them on the web page, and show the names of the classes with the top three confidences. There is an accompanying word cloud, where the size of the word is correlated to the frequency of the class being the top choice for a picture.
Ready to put this code pattern to use? Complete details on how to get started running and using this application are in the README.
Use computer vision, TensorFlow, and Keras for image classification and processing.
Artificial IntelligenceData Science+
Improve your neural network model by using some well-known machine learning techniques.
Learn how you can integrate Watson Assistant into your LINE app to create a chatbot that's easy to understand and…
Artificial IntelligenceDeep Learning+
Back to top