fixed:'str' object has no attribute 'base_dtype'

This commit is contained in:
Zchen
2025-10-15 23:11:02 +08:00
parent e8f0308fef
commit 83621f91f0

View File

@@ -52,13 +52,13 @@ class NoiseModel(keras.Model):
weight = self.add_weight( weight = self.add_weight(
name=f'day_weight_{i}', name=f'day_weight_{i}',
shape=(neural_dim, neural_dim), shape=(neural_dim, neural_dim),
initializer='identity', initializer=tf.keras.initializers.Identity(),
trainable=True trainable=True
) )
bias = self.add_weight( bias = self.add_weight(
name=f'day_bias_{i}', name=f'day_bias_{i}',
shape=(neural_dim,), shape=(neural_dim,),
initializer='zeros', initializer=tf.keras.initializers.Zeros(),
trainable=True trainable=True
) )
self.day_weights.append(weight) self.day_weights.append(weight)
@@ -220,13 +220,13 @@ class CleanSpeechModel(keras.Model):
weight = self.add_weight( weight = self.add_weight(
name=f'day_weight_{i}', name=f'day_weight_{i}',
shape=(neural_dim, neural_dim), shape=(neural_dim, neural_dim),
initializer='identity', initializer=tf.keras.initializers.Identity(),
trainable=True trainable=True
) )
bias = self.add_weight( bias = self.add_weight(
name=f'day_bias_{i}', name=f'day_bias_{i}',
shape=(neural_dim,), shape=(neural_dim,),
initializer='zeros', initializer=tf.keras.initializers.Zeros(),
trainable=True trainable=True
) )
self.day_weights.append(weight) self.day_weights.append(weight)