From 615a5e8d463e1ad05f05afe9dfcdbdf6d81c29be Mon Sep 17 00:00:00 2001 From: jinzr Date: Thu, 30 Nov 2023 22:41:46 +0800 Subject: [PATCH] Update tts_datamodule.py --- egs/vctk/TTS/vits/tts_datamodule.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/egs/vctk/TTS/vits/tts_datamodule.py b/egs/vctk/TTS/vits/tts_datamodule.py index f7772d6d6..37534f510 100644 --- a/egs/vctk/TTS/vits/tts_datamodule.py +++ b/egs/vctk/TTS/vits/tts_datamodule.py @@ -176,6 +176,7 @@ class VctkTtsDataModule: train = SpeechSynthesisDataset( return_text=False, return_tokens=True, + return_spk_ids=True, feature_input_strategy=eval(self.args.input_strategy)(), return_cuts=self.args.return_cuts, ) @@ -191,6 +192,7 @@ class VctkTtsDataModule: train = SpeechSynthesisDataset( return_text=False, return_tokens=True, + return_spk_ids=True, feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), return_cuts=self.args.return_cuts, ) @@ -246,6 +248,7 @@ class VctkTtsDataModule: validate = SpeechSynthesisDataset( return_text=False, return_tokens=True, + return_spk_ids=True, feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), return_cuts=self.args.return_cuts, ) @@ -253,6 +256,7 @@ class VctkTtsDataModule: validate = SpeechSynthesisDataset( return_text=False, return_tokens=True, + return_spk_ids=True, feature_input_strategy=eval(self.args.input_strategy)(), return_cuts=self.args.return_cuts, ) @@ -285,6 +289,7 @@ class VctkTtsDataModule: test = SpeechSynthesisDataset( return_text=False, return_tokens=True, + return_spk_ids=True, feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), return_cuts=self.args.return_cuts, ) @@ -292,6 +297,7 @@ class VctkTtsDataModule: test = SpeechSynthesisDataset( return_text=False, return_tokens=True, + return_spk_ids=True, feature_input_strategy=eval(self.args.input_strategy)(), return_cuts=self.args.return_cuts, )