lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change optimizer in CIFAR10 example.

Browse Source
This commit is contained in:
Francois Chollet 2016-12-12 16:02:43 -08:00
parent 7e2e7a5e5a
commit 7c4f033c6a
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/cifar10_cnn.py
View File

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