From 2a8b62d7d298898a40f2a3a8f4894e6d63ca03c5 Mon Sep 17 00:00:00 2001 From: WongLaw Date: Mon, 9 Jan 2023 06:52:54 +0000 Subject: [PATCH] vits_init, test=tts --- paddlespeech/t2s/models/vits/text_encoder.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddlespeech/t2s/models/vits/text_encoder.py b/paddlespeech/t2s/models/vits/text_encoder.py index 47b8b8c65..5fad3d8ef 100644 --- 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,