1515iterations = 391
1616num_classes = 10
1717dropout = 0.5
18+ weight_decay = 0.0001
1819log_filepath = './nin_bn'
1920
2021def color_preprocessing (x_train ,x_test ):
@@ -40,39 +41,39 @@ def scheduler(epoch):
4041def build_model ():
4142 model = Sequential ()
4243
43- model .add (Conv2D (192 , (5 , 5 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.01 ) , input_shape = x_train .shape [1 :]))
44+ model .add (Conv2D (192 , (5 , 5 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" , input_shape = x_train .shape [1 :]))
4445 model .add (BatchNormalization ())
4546 model .add (Activation ('relu' ))
46- model .add (Conv2D (160 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
47+ model .add (Conv2D (160 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
4748 model .add (BatchNormalization ())
4849 model .add (Activation ('relu' ))
49- model .add (Conv2D (96 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
50+ model .add (Conv2D (96 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
5051 model .add (BatchNormalization ())
5152 model .add (Activation ('relu' ))
5253 model .add (MaxPooling2D (pool_size = (3 , 3 ),strides = (2 ,2 ),padding = 'same' ))
5354
5455 model .add (Dropout (dropout ))
5556
56- model .add (Conv2D (192 , (5 , 5 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
57+ model .add (Conv2D (192 , (5 , 5 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
5758 model .add (BatchNormalization ())
5859 model .add (Activation ('relu' ))
59- model .add (Conv2D (192 , (1 , 1 ),padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
60+ model .add (Conv2D (192 , (1 , 1 ),padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
6061 model .add (BatchNormalization ())
6162 model .add (Activation ('relu' ))
62- model .add (Conv2D (192 , (1 , 1 ),padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
63+ model .add (Conv2D (192 , (1 , 1 ),padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
6364 model .add (BatchNormalization ())
6465 model .add (Activation ('relu' ))
6566 model .add (MaxPooling2D (pool_size = (3 , 3 ),strides = (2 ,2 ),padding = 'same' ))
6667
6768 model .add (Dropout (dropout ))
6869
69- model .add (Conv2D (192 , (3 , 3 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
70+ model .add (Conv2D (192 , (3 , 3 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
7071 model .add (BatchNormalization ())
7172 model .add (Activation ('relu' ))
72- model .add (Conv2D (192 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
73+ model .add (Conv2D (192 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
7374 model .add (BatchNormalization ())
7475 model .add (Activation ('relu' ))
75- model .add (Conv2D (10 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (0.0001 ), kernel_initializer = RandomNormal ( stddev = 0.05 ) ))
76+ model .add (Conv2D (10 , (1 , 1 ), padding = 'same' , kernel_regularizer = keras .regularizers .l2 (weight_decay ), kernel_initializer = "he_normal" ))
7677 model .add (BatchNormalization ())
7778 model .add (Activation ('relu' ))
7879
0 commit comments