mirror of https://github.com/coqui-ai/TTS.git
Fix #616
This commit is contained in:
parent
676d22ff0e
commit
3c0454490f
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue