mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-05 15:14:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update tts_datamodule.py

Browse Source
This commit is contained in:
jinzr 2023-11-30 22:41:46 +08:00
parent 523635cf77
commit 615a5e8d46
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/vctk/TTS/vits/tts_datamodule.py
View File

@ -176,6 +176,7 @@ class VctkTtsDataModule:
train = SpeechSynthesisDataset( train = SpeechSynthesisDataset(
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
return_spk_ids=True,
feature_input_strategy=eval(self.args.input_strategy)(), feature_input_strategy=eval(self.args.input_strategy)(),
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )
@ -191,6 +192,7 @@ class VctkTtsDataModule:
train = SpeechSynthesisDataset( train = SpeechSynthesisDataset(
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
return_spk_ids=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )
@ -246,6 +248,7 @@ class VctkTtsDataModule:
validate = SpeechSynthesisDataset( validate = SpeechSynthesisDataset(
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
return_spk_ids=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )
@ -253,6 +256,7 @@ class VctkTtsDataModule:
validate = SpeechSynthesisDataset( validate = SpeechSynthesisDataset(
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
return_spk_ids=True,
feature_input_strategy=eval(self.args.input_strategy)(), feature_input_strategy=eval(self.args.input_strategy)(),
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )
@ -285,6 +289,7 @@ class VctkTtsDataModule:
test = SpeechSynthesisDataset( test = SpeechSynthesisDataset(
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
return_spk_ids=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )
@ -292,6 +297,7 @@ class VctkTtsDataModule:
test = SpeechSynthesisDataset( test = SpeechSynthesisDataset(
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
return_spk_ids=True,
feature_input_strategy=eval(self.args.input_strategy)(), feature_input_strategy=eval(self.args.input_strategy)(),
return_cuts=self.args.return_cuts, return_cuts=self.args.return_cuts,
) )