From f0616a3b2473c3d0196289cd0a1c5be9a120845a Mon Sep 17 00:00:00 2001 From: erogol Date: Mon, 8 Jun 2020 10:45:38 +0200 Subject: [PATCH] linter fix --- models/tacotron2.py | 2 +- train.py | 2 +- vocoder/train.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/models/tacotron2.py b/models/tacotron2.py index bce21e9e..46b915b5 100644 --- a/models/tacotron2.py +++ b/models/tacotron2.py @@ -130,7 +130,7 @@ class Tacotron2(TacotronAbstract): self.compute_speaker_embedding(speaker_ids) if self.num_speakers > 1: encoder_outputs = self._add_speaker_embedding(encoder_outputs, - self.speaker_embeddings) + self.speaker_embeddings) decoder_outputs, alignments, stop_tokens = self.decoder.inference( encoder_outputs) postnet_outputs = self.postnet(decoder_outputs) diff --git a/train.py b/train.py index a47828e0..a5f3d2e7 100644 --- a/train.py +++ b/train.py @@ -350,7 +350,7 @@ def evaluate(model, criterion, ap, global_step, epoch): loss_dict = criterion(postnet_output, decoder_output, mel_input, linear_input, stop_tokens, stop_targets, mel_lengths, decoder_backward_output, - alignments, alignment_lengths, alignments_backward, + alignments, alignment_lengths, alignments_backward, text_lengths) if c.bidirectional_decoder: keep_avg.update_values({'avg_decoder_b_loss': loss_dict['decoder_b_loss'].item(), diff --git a/vocoder/train.py b/vocoder/train.py index 54b36e6b..259c5eb7 100644 --- a/vocoder/train.py +++ b/vocoder/train.py @@ -519,7 +519,7 @@ def main(args): # pylint: disable=redefined-outer-name model_disc, criterion_disc, optimizer_disc, scheduler_gen, scheduler_disc, ap, global_step, 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) c_logger.print_epoch_end(epoch, eval_avg_loss_dict) target_loss = eval_avg_loss_dict[c.target_loss]