storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-11-02 18:09:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add max_decoder_steps, streamline Dockerfile

Browse Source
This commit is contained in:
Brandon Thomas
2022-11-17 21:38:14 -05:00
parent 5163650b86
commit c7a880924f
5 changed files with 144 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/tacotron2/model.py
+3
View File
@@ -469,6 +469,9 @@ class Tacotron2(nn.Module):
self.encoder = Encoder(hparams)
self.decoder = Decoder(hparams)
self.postnet = Postnet(hparams)
def fakeyou_set_max_decoder_steps(self, max_decoder_steps):
self.decoder.max_decoder_steps = max_decoder_steps
def parse_batch(self, batch):
text_padded, input_lengths, mel_padded, gate_padded, \