Change number of filters of second conv layer to 16

vmelan · Jul 24, 2018 · 561c82e · 561c82e
1 parent 580015d
commit 561c82e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/LeNet/Keras/model.py b/LeNet/Keras/model.py
@@ -30,7 +30,7 @@ def build_model(self, input_shape=(32, 32, 1)):
 		X = Activation("tanh", name="Tanh_1")(X)
 		X = AveragePooling2D(pool_size=(2, 2), name="AvgPool_1")(X)
 		# 2nd conv layer : CONV + TANH + AVERAGE POOL
-		X = Conv2D(filters=6, kernel_size=(5, 5), strides=(1, 1), \
+		X = Conv2D(filters=16, kernel_size=(5, 5), strides=(1, 1), \
 			kernel_initializer=glorot_uniform(seed=0), \
 			padding="valid", \
 			name="Conv_2")(X)