From 83621f91f0ecb283a2db1bd9b4454f723e4758d7 Mon Sep 17 00:00:00 2001
From: Zchen <161216199+ZH-CEN@users.noreply.github.com>
Date: Wed, 15 Oct 2025 23:11:02 +0800
Subject: [PATCH] fixed:'str' object has no attribute 'base_dtype'

---
 model_training_nnn_tpu/rnn_model_tf.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/model_training_nnn_tpu/rnn_model_tf.py b/model_training_nnn_tpu/rnn_model_tf.py
index 15fff17..14dd242 100644
--- a/model_training_nnn_tpu/rnn_model_tf.py
+++ b/model_training_nnn_tpu/rnn_model_tf.py
@@ -52,13 +52,13 @@ class NoiseModel(keras.Model):
             weight = self.add_weight(
                 name=f'day_weight_{i}',
                 shape=(neural_dim, neural_dim),
-                initializer='identity',
+                initializer=tf.keras.initializers.Identity(),
                 trainable=True
             )
             bias = self.add_weight(
                 name=f'day_bias_{i}',
                 shape=(neural_dim,),
-                initializer='zeros',
+                initializer=tf.keras.initializers.Zeros(),
                 trainable=True
             )
             self.day_weights.append(weight)
@@ -220,13 +220,13 @@ class CleanSpeechModel(keras.Model):
             weight = self.add_weight(
                 name=f'day_weight_{i}',
                 shape=(neural_dim, neural_dim),
-                initializer='identity',
+                initializer=tf.keras.initializers.Identity(),
                 trainable=True
             )
             bias = self.add_weight(
                 name=f'day_bias_{i}',
                 shape=(neural_dim,),
-                initializer='zeros',
+                initializer=tf.keras.initializers.Zeros(),
                 trainable=True
             )
             self.day_weights.append(weight)