Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update docstrings

This commit is contained in:
Eren Gölge 2021-07-22 14:20:43 +02:00
parent 57b3aec1b9
commit b81560607b
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/tts/layers/glow_tts/monotonic_align/__init__.py
View File

@ -21,8 +21,10 @@ def convert_pad_shape(pad_shape):
def generate_path(duration, mask): def generate_path(duration, mask):
""" """
duration: [b, t_x] Shapes:
mask: [b, t_x, t_y] - duration: :math:`[B, T_en]`
- mask: :math:'[B, T_en, T_de]`
- path: :math:`[B, T_en, T_de]`
""" """
device = duration.device device = duration.device
b, t_x, t_y = mask.shape b, t_x, t_y = mask.shape

6
TTS/tts/layers/losses.py
View File

@ -69,9 +69,9 @@ class MSELossMasked(nn.Module):
length: A Variable containing a LongTensor of size (batch,) length: A Variable containing a LongTensor of size (batch,)
which contains the length of each data in a batch. which contains the length of each data in a batch.
Shapes: Shapes:
x: B x T X D - x: :math:`[B, T, D]`
target: B x T x D - target: :math:`[B, T, D]`
length: B - length: :math:`B`
Returns: Returns:
loss: An average loss value in range [0, 1] masked by the length. loss: An average loss value in range [0, 1] masked by the length.
""" """