changed to channels first

llvm/projects/keras/src/mobilenet_cifar10.py

@@ -11,7 +11,7 @@ from keras.models import Model
 from keras import optimizers
 import keras.backend as K
 
-K.set_image_dim_ordering('th')
+K.set_image_data_format('channels_first')
 
 
 (X_train, y_train), (X_test, y_test) = cifar10.load_data()