fix bugs in libritts training

yerfor · May 13, 2022 · d4f300e · d4f300e
1 parent 0ffda74
commit d4f300e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/egs/datasets/audio/libritts/base_text2mel.yaml b/egs/datasets/audio/libritts/base_text2mel.yaml
@@ -19,4 +19,5 @@ f0_min: 80
 f0_max: 600
 vocoder: PWG
 vocoder_ckpt: checkpoints/pwg_libritts
-num_spk: 2000
+num_spk: 2000
+use_spk_id: true
diff --git a/modules/tts/syntaspeech/syntaspeech.py b/modules/tts/syntaspeech/syntaspeech.py
@@ -109,8 +109,10 @@ def forward(self, txt_tokens, word_tokens, ph2word, word_len, mel2word=None, mel
                 spk_embed=None, spk_id=None, pitch=None, infer=False, tgt_mels=None,
                 global_step=None, graph_lst=None, etypes_lst=None, *args, **kwargs):     
 
-        if self.hparams['use_spk_embed'] or self.hparams['use_spk_id']:
-            spk_embed = self.spk_embed_proj(spk_embed)[:, None, :]  
+        if self.hparams['use_spk_embed']:
+            spk_embed = spk_embed
+        elif self.hparams['use_spk_id']:
+            spk_embed = self.spk_embed_proj(spk_id)[:, None, :]  
         else:
             spk_embed = 0