From 4c44385bad4f594186a83de984e47fb78a4b3c03 Mon Sep 17 00:00:00 2001 From: Felipe Honorato Date: Sat, 13 Apr 2024 12:03:55 -0300 Subject: [PATCH] Fixing l2 regularizer param --- deep_speaker/conv_models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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):