diff --git a/examples/cifar10_cnn.py b/examples/cifar10_cnn.py index bbb774aac..f81034954 100644 --- a/examples/cifar10_cnn.py +++ b/examples/cifar10_cnn.py @@ -13,7 +13,6 @@ from keras.preprocessing.image import ImageDataGenerator from keras.models import Sequential from keras.layers import Dense, Dropout, Activation, Flatten from keras.layers import Convolution2D, MaxPooling2D -from keras.optimizers import SGD from keras.utils import np_utils batch_size = 32 @@ -60,10 +59,9 @@ model.add(Dropout(0.5)) model.add(Dense(nb_classes)) model.add(Activation('softmax')) -# Let's train the model using SGD + momentum: -sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True) +# Let's train the model using RMSprop model.compile(loss='categorical_crossentropy', - optimizer=sgd, + optimizer='rmsprop', metrics=['accuracy']) X_train = X_train.astype('float32')