From de7855665507bb792df91dab439bc9d3cd18e3d9 Mon Sep 17 00:00:00 2001 From: Edresson Date: Sat, 28 Aug 2021 09:57:52 -0300 Subject: [PATCH] Fix the optimizer parameters bug in multilingual and multispeaker training --- TTS/tts/models/vits.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 600a9551..d355d5c1 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -882,8 +882,12 @@ class Vits(BaseTTS): self.waveform_decoder.parameters(), ) # add the speaker embedding layer - if hasattr(self, "emb_g"): + if hasattr(self, "emb_g") and self.args.use_speaker_embedding and not self.args.use_d_vector_file: gen_parameters = chain(gen_parameters, self.emb_g.parameters()) + # add the language embedding layer + if hasattr(self, "emb_l") and self.args.use_language_embedding: + gen_parameters = chain(gen_parameters, self.emb_l.parameters()) + optimizer0 = get_optimizer( self.config.optimizer, self.config.optimizer_params, self.config.lr_gen, parameters=gen_parameters )