From 3c0454490f048f9f32b201c5efa12b4af57c02c4 Mon Sep 17 00:00:00 2001 From: eren golge Date: Tue, 6 Jul 2021 11:05:05 +0200 Subject: [PATCH] Fix #616 --- TTS/tts/models/base_tacotron.py | 2 -- recipes/ljspeech/tacotron2-DDC/tacotron2-DDC.json | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/TTS/tts/models/base_tacotron.py b/TTS/tts/models/base_tacotron.py index a99e1926..b7056e06 100644 --- a/TTS/tts/models/base_tacotron.py +++ b/TTS/tts/models/base_tacotron.py @@ -281,6 +281,4 @@ class BaseTacotron(BaseTTS): self.decoder.set_r(r) if trainer.config.bidirectional_decoder: trainer.model.decoder_backward.set_r(r) - trainer.train_loader = trainer.setup_train_dataloader(self.ap, self.model.decoder.r, verbose=True) - trainer.eval_loader = trainer.setup_eval_dataloder(self.ap, self.model.decoder.r) print(f"\n > Number of output frames: {self.decoder.r}") diff --git a/recipes/ljspeech/tacotron2-DDC/tacotron2-DDC.json b/recipes/ljspeech/tacotron2-DDC/tacotron2-DDC.json index e3531851..d787c138 100644 --- a/recipes/ljspeech/tacotron2-DDC/tacotron2-DDC.json +++ b/recipes/ljspeech/tacotron2-DDC/tacotron2-DDC.json @@ -1,4 +1,5 @@ { + "model": "Tacotron2", "datasets": [ { "name": "ljspeech", @@ -38,7 +39,6 @@ }, "distributed_backend": "gloo", "distributed_url": "tcp:\/\/localhost:54321", - "model": "Tacotron2", "run_name": "ljspeech-ddc", "run_description": "tacotron2 with double decoder consistency.", "batch_size": 64,