code format, test=tts

3 years ago · c6e6210964
parent 88adcaa6dc
commit c6e6210964
1 changed files with 3 additions and 3 deletions
--- a/paddlespeech/t2s/modules/predictor/length_regulator.py
+++ b/paddlespeech/t2s/modules/predictor/length_regulator.py
@ -49,9 +49,9 @@ class LengthRegulator(nn.Layer):
        encodings: (B, T, C)
        durations: (B, T)
        """
-        #batch_size, t_enc = durations.shape
+        #batch_size, t_enc = durations.shape  # linux
-        batch_size = paddle.shape(durations)[0]
+        batch_size = paddle.shape(durations)[0]  # windows and mac
-        t_enc = paddle.shape(durations)[1]
+        t_enc = paddle.shape(durations)[1]  # windows and mac
        durations = durations.numpy()
        slens = np.sum(durations, -1)
        t_dec = np.max(slens)