Skip to content

Commit 7416e7d

Browse files
committed
commit #40: change weight int -> he normal for general
1 parent ee73495 commit 7416e7d

File tree

2 files changed

+24
-24
lines changed

2 files changed

+24
-24
lines changed

2_Network_in_Network/Network_in_Network_bn_keras.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
iterations = 391
1616
num_classes = 10
1717
dropout = 0.5
18+
weight_decay = 0.0001
1819
log_filepath = './nin_bn'
1920

2021
def color_preprocessing(x_train,x_test):
@@ -40,39 +41,39 @@ def scheduler(epoch):
4041
def 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

2_Network_in_Network/Network_in_Network_keras.py

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
iterations = 391
1616
num_classes = 10
1717
dropout = 0.5
18+
weight_decay = 0.0001
1819
log_filepath = './nin'
1920

2021
def color_preprocessing(x_train,x_test):
@@ -29,42 +30,40 @@ def color_preprocessing(x_train,x_test):
2930
return x_train, x_test
3031

3132
def scheduler(epoch):
32-
if epoch <= 60:
33-
return 0.05
34-
if epoch <= 120:
33+
if epoch <= 80:
3534
return 0.01
36-
if epoch <= 160:
37-
return 0.002
38-
return 0.0004
35+
if epoch <= 140:
36+
return 0.005
37+
return 0.001
3938

4039
def build_model():
4140
model = Sequential()
4241

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:]))
42+
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:]))
4443
model.add(Activation('relu'))
45-
model.add(Conv2D(160, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
44+
model.add(Conv2D(160, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
4645
model.add(Activation('relu'))
47-
model.add(Conv2D(96, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
46+
model.add(Conv2D(96, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
4847
model.add(Activation('relu'))
4948
model.add(MaxPooling2D(pool_size=(3, 3),strides=(2,2),padding = 'same'))
5049

5150
model.add(Dropout(dropout))
5251

53-
model.add(Conv2D(192, (5, 5), padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
52+
model.add(Conv2D(192, (5, 5), padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
5453
model.add(Activation('relu'))
55-
model.add(Conv2D(192, (1, 1),padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
54+
model.add(Conv2D(192, (1, 1),padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
5655
model.add(Activation('relu'))
57-
model.add(Conv2D(192, (1, 1),padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
56+
model.add(Conv2D(192, (1, 1),padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
5857
model.add(Activation('relu'))
5958
model.add(MaxPooling2D(pool_size=(3, 3),strides=(2,2),padding = 'same'))
6059

6160
model.add(Dropout(dropout))
6261

63-
model.add(Conv2D(192, (3, 3), padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
62+
model.add(Conv2D(192, (3, 3), padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
6463
model.add(Activation('relu'))
65-
model.add(Conv2D(192, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
64+
model.add(Conv2D(192, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
6665
model.add(Activation('relu'))
67-
model.add(Conv2D(10, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(0.0001), kernel_initializer=RandomNormal(stddev = 0.05)))
66+
model.add(Conv2D(10, (1, 1), padding='same', kernel_regularizer=keras.regularizers.l2(weight_decay), kernel_initializer="he_normal"))
6867
model.add(Activation('relu'))
6968

7069
model.add(GlobalAveragePooling2D())

0 commit comments

Comments
 (0)