diff --git a/deep_speaker/conv_models.py b/deep_speaker/conv_models.py index b7087ee..988853a 100644 --- a/deep_speaker/conv_models.py +++ b/deep_speaker/conv_models.py @@ -186,7 +186,7 @@ def identity_block(self, input_tensor, kernel_size, filters, stage, block): activation=None, padding='same', kernel_initializer='glorot_uniform', - kernel_regularizer=regularizers.l2(l=0.0001), + kernel_regularizer=regularizers.l2(l2=0.0001), name=conv_name_base + '_2a')(input_tensor) x = BatchNormalization(name=conv_name_base + '_2a_bn')(x) x = self.clipped_relu(x) @@ -198,7 +198,7 @@ def identity_block(self, input_tensor, kernel_size, filters, stage, block): activation=None, padding='same', kernel_initializer='glorot_uniform', - kernel_regularizer=regularizers.l2(l=0.0001), + kernel_regularizer=regularizers.l2(l2=0.0001), name=conv_name_base + '_2b', )(x) x = BatchNormalization(name=conv_name_base + '_2b_bn')(x) @@ -218,7 +218,7 @@ def conv_and_res_block(self, inp, filters, stage): activation=None, padding='same', kernel_initializer='glorot_uniform', - kernel_regularizer=regularizers.l2(l=0.0001), name=conv_name)(inp) + kernel_regularizer=regularizers.l2(l2=0.0001), name=conv_name)(inp) o = BatchNormalization(name=conv_name + '_bn')(o) o = self.clipped_relu(o) for i in range(3):