From 05143308690ce24a208a0704ff4719f7b5ea1869 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Mon, 15 Mar 2021 11:28:06 +0100
Subject: [PATCH] fix mozilla/TTS#685

---
 TTS/tts/models/tacotron_abstract.py |  3 +--
 tests/test_tacotron_train.sh        | 11 +++++++++++
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/TTS/tts/models/tacotron_abstract.py b/TTS/tts/models/tacotron_abstract.py
index 10953269..22e86ee4 100644
--- a/TTS/tts/models/tacotron_abstract.py
+++ b/TTS/tts/models/tacotron_abstract.py
@@ -149,8 +149,7 @@ class TacotronAbstract(ABC, nn.Module):
     def _backward_pass(self, mel_specs, encoder_outputs, mask):
         """ Run backwards decoder """
         decoder_outputs_b, alignments_b, _ = self.decoder_backward(
-            encoder_outputs, torch.flip(mel_specs, dims=(1,)), mask,
-            self.speaker_embeddings_projected)
+            encoder_outputs, torch.flip(mel_specs, dims=(1,)), mask)
         decoder_outputs_b = decoder_outputs_b.transpose(1, 2).contiguous()
         return decoder_outputs_b, alignments_b
 
diff --git a/tests/test_tacotron_train.sh b/tests/test_tacotron_train.sh
index fa9930a7..e0a0253b 100755
--- a/tests/test_tacotron_train.sh
+++ b/tests/test_tacotron_train.sh
@@ -2,6 +2,7 @@
 set -xe
 BASEDIR=$(dirname "$0")
 echo "$BASEDIR"
+
 # run training
 CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/inputs/test_tacotron_config.json
 # find the training folder
@@ -12,6 +13,16 @@ CUDA_VISIBLE_DEVICES=""  python TTS/bin/train_tacotron.py --continue_path $BASED
 # remove all the outputs
 rm -rf $BASEDIR/train_outputs/
 
+# run Tacotron bi-directional decoder
+CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/inputs/test_tacotron_bd_config.json
+# find the training folder
+LATEST_FOLDER=$(ls $BASEDIR/train_outputs/| sort | tail -1)
+echo $LATEST_FOLDER
+# continue the previous training
+CUDA_VISIBLE_DEVICES=""  python TTS/bin/train_tacotron.py --continue_path $BASEDIR/train_outputs/$LATEST_FOLDER
+# remove all the outputs
+rm -rf $BASEDIR/train_outputs/
+
 # Tacotron2
 # run training
 CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/inputs/test_tacotron2_config.json