This commit is contained in:
eren golge
2021-07-06 14:44:03 +02:00
committed by Eren Gölge
parent 676d22ff0e
commit 3c0454490f
2 changed files with 1 additions and 3 deletions
-2
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}")
@@ -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,