relu

miquelmarti · May 23, 2016 · 5b431db · 5b431db
1 parent 7c98f14
commit 5b431db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tiny_imagenet_classifier.py b/tiny_imagenet_classifier.py
@@ -190,7 +190,8 @@
     model.add(Convolution2D(192, 1, 1, border_mode='same'))
     model.add(Activation('relu'))
     model.add(Convolution2D(10, 1, 1, border_mode='same'))
-    AveragePooling2D(pool_size=(6, 6), strides=None, border_mode='valid', dim_ordering='th')
+    model.add(Activation('relu'))
+    AveragePooling2D(pool_size=(6, 6), strides=None, border_mode='same', dim_ordering='th')
     # model.add(Activation('relu'))
     model.add(Flatten())
     # model.add(Dense(512))