vits_init, test=tts

3 years ago · 2a8b62d7d2
parent 956a8acbc5
commit 2a8b62d7d2
1 changed files with 2 additions and 0 deletions
--- a/paddlespeech/t2s/models/vits/text_encoder.py
+++ b/paddlespeech/t2s/models/vits/text_encoder.py
@ -131,6 +131,8 @@ class TextEncoder(nn.Layer):
            cnn_module_kernel=conformer_kernel_size, )
        self.proj = nn.Conv1D(attention_dim, attention_dim * 2, 1)

+        self.reset_parameters()
+        
    def forward(
            self,
            x: paddle.Tensor,