Fix cifar10 example

2015-04-16 00:50:32 -07:00 · 2015-04-16 00:50:32 -07:00 · 5f25e2205a
commit 5f25e2205a
parent cb02df95b2
1 changed files with 1 additions and 1 deletions
--- a/examples/cifar10_cnn.py
+++ b/examples/cifar10_cnn.py
@ -53,7 +53,7 @@ model.add(Activation('relu'))
 model.add(MaxPooling2D(poolsize=(2, 2)))
 model.add(Dropout(0.25))

-model.add(Flatten(64*8*8))
+model.add(Flatten())
 model.add(Dense(64*8*8, 512, init='normal'))
 model.add(Activation('relu'))
 model.add(Dropout(0.5))