make style and linter fixes

This commit is contained in:
Eren Gölge 2021-06-03 13:05:54 +02:00
parent a605dd3d08
commit 147550c65f
2 changed files with 17 additions and 7 deletions

View File

@ -81,7 +81,6 @@ def format_data(data):
# setup input data
text_input = data[0]
text_lengths = data[1]
speaker_names = data[2]
mel_input = data[4]
mel_lengths = data[5]
item_idx = data[7]

View File

@ -268,7 +268,7 @@ class TrainerTTS:
data_items: List,
verbose: bool,
speaker_ids: Union[Dict, List],
d_vectors: Union[Dict, List]
d_vectors: Union[Dict, List],
) -> DataLoader:
if is_eval and not self.config.run_eval:
loader = None
@ -290,10 +290,10 @@ class TrainerTTS:
enable_eos_bos=self.config.enable_eos_bos_chars,
use_noise_augment=not is_eval,
verbose=verbose,
speaker_id_mapping=speaker_ids
if self.config.use_speaker_embedding else None,
speaker_id_mapping=speaker_ids if self.config.use_speaker_embedding else None,
d_vector_mapping=d_vectors
if self.config.use_speaker_embedding and self.config.use_external_speaker_embedding_file else None,
if self.config.use_speaker_embedding and self.config.use_external_speaker_embedding_file
else None,
)
if self.config.use_phonemes and self.config.compute_input_seq_cache:
@ -383,6 +383,7 @@ class TrainerTTS:
return {
"text_input": text_input,
"text_lengths": text_lengths,
"speaker_names": speaker_names,
"mel_input": mel_input,
"mel_lengths": mel_lengths,
"linear_input": linear_input,
@ -630,11 +631,21 @@ class TrainerTTS:
# define data loaders
self.train_loader = self.get_train_dataloader(
self.config.r, self.ap, self.data_train, verbose=True, speaker_ids=self.speaker_manager.speaker_ids, d_vectors=self.speaker_manager.d_vectors
self.config.r,
self.ap,
self.data_train,
verbose=True,
speaker_ids=self.speaker_manager.speaker_ids,
d_vectors=self.speaker_manager.d_vectors,
)
self.eval_loader = (
self.get_eval_dataloder(
self.config.r, self.ap, self.data_train, verbose=True, speaker_ids=self.speaker_manager.speaker_ids, d_vectors=self.speaker_manager.d_vectors
self.config.r,
self.ap,
self.data_train,
verbose=True,
speaker_ids=self.speaker_manager.speaker_ids,
d_vectors=self.speaker_manager.d_vectors,
)
if self.config.run_eval
else None