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.
March 14, 2019
March 29, 2019
Artificial intelligenceData Science+
June 11, 2019
Back to top