Made r value in quick_start.py for tts_model no longer hardcoded

scarlettlin2022 · Sep 5, 2019 · 31179a6 · 31179a6
1 parent fb8fcb4
commit 31179a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/quick_start.py b/quick_start.py
@@ -100,8 +100,7 @@
     voc_k = voc_model.get_step() // 1000
     tts_k = tts_model.get_step() // 1000
 
-    # TODO: get rid of this hardcoding
-    r = 2
+    r = tts_model.r
 
     simple_table([('WaveRNN', str(voc_k) + 'k'),
                   (f'Tacotron(r={r})', str(tts_k) + 'k'),