mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 06:34:20 +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")
train = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False,
return_tokens=True,
feature_input_strategy=eval(self.args.input_strategy)(),
@ -184,7 +183,6 @@ class LJSpeechTtsDataModule:
use_fft_mag=True,
)
train = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False,
return_tokens=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
@ -240,7 +238,6 @@ class LJSpeechTtsDataModule:
use_fft_mag=True,
)
validate = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False,
return_tokens=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
@ -248,7 +245,6 @@ class LJSpeechTtsDataModule:
)
else:
validate = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False,
return_tokens=True,
feature_input_strategy=eval(self.args.input_strategy)(),
@ -281,7 +277,6 @@ class LJSpeechTtsDataModule:
use_fft_mag=True,
)
test = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False,
return_tokens=True,
feature_input_strategy=OnTheFlyFeatures(Spectrogram(config)),
@ -289,7 +284,6 @@ class LJSpeechTtsDataModule:
)
else:
test = SpeechSynthesisDataset(
return_token_ids=False,
return_text=False,
return_tokens=True,
feature_input_strategy=eval(self.args.input_strategy)(),