testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modified docstring of tacotron2

This commit is contained in:
lfchener 2020-12-18 20:28:21 +08:00
parent 63285dc80f
commit 255ddcfe32
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
parakeet/models/tacotron2.py
View File

@ -752,16 +752,13 @@ class Tacotron2(nn.Layer):
config: yacs.config.CfgNode config: yacs.config.CfgNode
Model configs. Model configs.
checkpoint_path: Path checkpoint_path: Path or str
The path of pretrained model checkpoint. The path of pretrained model checkpoint, without extension name.
Returns Returns
------- -------
mel_outputs_postnet: Tensor [shape=(T_mel, C)] Tacotron2
Output sequence of sepctrogram after postnet. The model build from pretrined result.
alignments: Tensor [shape=(T_mel, T_text)]
Attention weights.
""" """
model = cls(frontend, model = cls(frontend,
d_mels=config.data.d_mels, d_mels=config.data.d_mels,