zchen/b2txt25
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Zchen
2025-10-15 23:21:06 +08:00
parent 6c400a066c
commit ec8509ad31
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
model_training_nnn_tpu/rnn_model_tf.py
View File

@@ -79,8 +79,8 @@ class NoiseModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, # Avoid recurrent dropout on TPU recurrent_dropout=0.0, # Avoid recurrent dropout on TPU
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='noise_gru1' name='noise_gru1'
) )
@@ -90,8 +90,8 @@ class NoiseModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, recurrent_dropout=0.0,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='noise_gru2' name='noise_gru2'
) )
@@ -246,8 +246,8 @@ class CleanSpeechModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, recurrent_dropout=0.0,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='clean_gru1' name='clean_gru1'
) )
@@ -257,8 +257,8 @@ class CleanSpeechModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, recurrent_dropout=0.0,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='clean_gru2' name='clean_gru2'
) )
@@ -268,15 +268,15 @@ class CleanSpeechModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, recurrent_dropout=0.0,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='clean_gru3' name='clean_gru3'
) )
# Output classification layer # Output classification layer
self.output_layer = layers.Dense( self.output_layer = layers.Dense(
n_classes, n_classes,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
name='clean_output' name='clean_output'
) )
@@ -407,8 +407,8 @@ class NoisySpeechModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, recurrent_dropout=0.0,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='noisy_gru1' name='noisy_gru1'
) )
@@ -418,15 +418,15 @@ class NoisySpeechModel(keras.Model):
return_state=True, return_state=True,
dropout=self.rnn_dropout, dropout=self.rnn_dropout,
recurrent_dropout=0.0, recurrent_dropout=0.0,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
recurrent_initializer='orthogonal', recurrent_initializer=tf.keras.initializers.Orthogonal(),
name='noisy_gru2' name='noisy_gru2'
) )
# Output classification layer # Output classification layer
self.output_layer = layers.Dense( self.output_layer = layers.Dense(
n_classes, n_classes,
kernel_initializer='glorot_uniform', kernel_initializer=tf.keras.initializers.GlorotUniform(),
name='noisy_output' name='noisy_output'
) )