Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

linter fix

This commit is contained in:
erogol 2020-06-08 10:45:38 +02:00
parent 55b8952c3d
commit f0616a3b24
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/tacotron2.py
View File

@ -130,7 +130,7 @@ class Tacotron2(TacotronAbstract):
self.compute_speaker_embedding(speaker_ids) self.compute_speaker_embedding(speaker_ids)
if self.num_speakers > 1: if self.num_speakers > 1:
encoder_outputs = self._add_speaker_embedding(encoder_outputs, encoder_outputs = self._add_speaker_embedding(encoder_outputs,
self.speaker_embeddings) self.speaker_embeddings)
decoder_outputs, alignments, stop_tokens = self.decoder.inference( decoder_outputs, alignments, stop_tokens = self.decoder.inference(
encoder_outputs) encoder_outputs)
postnet_outputs = self.postnet(decoder_outputs) postnet_outputs = self.postnet(decoder_outputs)

2
vocoder/train.py
View File

@ -519,7 +519,7 @@ def main(args): # pylint: disable=redefined-outer-name
model_disc, criterion_disc, optimizer_disc, model_disc, criterion_disc, optimizer_disc,
scheduler_gen, scheduler_disc, ap, global_step, scheduler_gen, scheduler_disc, ap, global_step,
epoch) epoch)
eval_avg_loss_dict = evaluate(model_gen, criterion_gen, model_disc, ap, eval_avg_loss_dict = evaluate(model_gen, criterion_gen, model_disc, criterion_disc, ap,
global_step, epoch) global_step, epoch)
c_logger.print_epoch_end(epoch, eval_avg_loss_dict) c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
target_loss = eval_avg_loss_dict[c.target_loss] target_loss = eval_avg_loss_dict[c.target_loss]