(self, dictionary, embed_dim)
| 86 | padding=hparams['ffn_padding'], kernel_size=hparams['predictor_kernel']) |
| 87 | |
| 88 | def build_embedding(self, dictionary, embed_dim): |
| 89 | num_embeddings = len(dictionary) |
| 90 | emb = Embedding(num_embeddings, embed_dim, self.padding_idx) |
| 91 | return emb |
| 92 | |
| 93 | def forward(self, txt_tokens, mel2ph=None, spk_embed=None, |
| 94 | ref_mels=None, f0=None, uv=None, energy=None, skip_decoder=False, |