Fix cifar10 example
This commit is contained in:
parent
cb02df95b2
commit
5f25e2205a
@ -53,7 +53,7 @@ model.add(Activation('relu'))
|
|||||||
model.add(MaxPooling2D(poolsize=(2, 2)))
|
model.add(MaxPooling2D(poolsize=(2, 2)))
|
||||||
model.add(Dropout(0.25))
|
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(Dense(64*8*8, 512, init='normal'))
|
||||||
model.add(Activation('relu'))
|
model.add(Activation('relu'))
|
||||||
model.add(Dropout(0.5))
|
model.add(Dropout(0.5))
|
||||||
|
Loading…
Reference in New Issue
Block a user