more realistic values

lucidrains · Jul 22, 2024 · 04b9c1c · 04b9c1c
1 parent 96e4f72
commit 04b9c1c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/train_example.py b/train_example.py
@@ -11,16 +11,16 @@
 
 duration_predictor = DurationPredictor(
     transformer = dict(
-        dim = 80,
-        depth = 2,
+        dim = 512,
+        depth = 6,
     )
 )
 
 e2tts = E2TTS(
     duration_predictor = duration_predictor,
     transformer = dict(
-        dim = 80,
-        depth = 4,
+        dim = 512,
+        depth = 12,
         skip_connect_type = 'concat'
     ),
 )