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

fix tts_datamodule.py

Browse Source
This commit is contained in:
yaozengwei 2023-11-21 16:33:54 +08:00
parent 8791a4efb0
commit 32931b7885
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -168,7 +168,6 @@ class LJSpeechTtsDataModule:
""" """
logging.info("About to create train dataset") logging.info("About to create train dataset")
train = SpeechSynthesisDataset( train = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
feature_input_strategy=eval(self.args.input_strategy)(), feature_input_strategy=eval(self.args.input_strategy)(),
@ -184,7 +183,6 @@ class LJSpeechTtsDataModule:
use_fft_mag=True, use_fft_mag=True,
) )
train = SpeechSynthesisDataset( train = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
@ -240,7 +238,6 @@ class LJSpeechTtsDataModule:
use_fft_mag=True, use_fft_mag=True,
) )
validate = SpeechSynthesisDataset( validate = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
@ -248,7 +245,6 @@ class LJSpeechTtsDataModule:
) )
else: else:
validate = SpeechSynthesisDataset( validate = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
feature_input_strategy=eval(self.args.input_strategy)(), feature_input_strategy=eval(self.args.input_strategy)(),
@ -281,7 +277,6 @@ class LJSpeechTtsDataModule:
use_fft_mag=True, use_fft_mag=True,
) )
test = SpeechSynthesisDataset( test = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)), feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
@ -289,7 +284,6 @@ class LJSpeechTtsDataModule:
) )
else: else:
test = SpeechSynthesisDataset( test = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False, return_text=False,
return_tokens=True, return_tokens=True,
feature_input_strategy=eval(self.args.input_strategy)(), feature_input_strategy=eval(self.args.input_strategy)(),