Update length_regulator.py

4 years ago · 7ae4f7221e
parent acfe2b9084
commit 7ae4f7221e
1 changed files with 2 additions and 2 deletions
--- a/paddlespeech/t2s/modules/predictor/length_regulator.py
+++ b/paddlespeech/t2s/modules/predictor/length_regulator.py
@ -70,8 +70,8 @@ class LengthRegulator(nn.Layer):
        ----------
        xs : Tensor
            Batch of sequences of char or phoneme embeddings (B, Tmax, D).
-        ds : LongTensor
-                Batch of durations of each frame (B, T).
+        ds : Tensor(int64)
+            Batch of durations of each frame (B, T).
        alpha : float, optional
            Alpha value to control speed of speech.