Fix input shape dimensions

stjordanis · Nov 4, 2016 · b511df7 · b511df7
1 parent 3142eaa
commit b511df7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/cifar_generator_cnn.py b/examples/cifar_generator_cnn.py
@@ -60,7 +60,7 @@ def model(datagen, X_train, Y_train, X_test, Y_test):
     model = Sequential()
 
     model.add(Convolution2D(32, 3, 3, border_mode='same',
-                            input_shape=(img_channels, img_rows, img_cols)))
+                            input_shape=X_train.shape[1:]))
     model.add(Activation('relu'))
     model.add(Convolution2D(32, 3, 3))
     model.add(Activation('relu'))