This commit is contained in:
eren golge 2021-07-06 11:05:05 +02:00 committed by Eren Gölge
parent 676d22ff0e
commit 3c0454490f
2 changed files with 1 additions and 3 deletions

View File

@ -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}")

View File

@ -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,