From a4c80be48bc56dfec26425bd49c0b69fb2cfd92c Mon Sep 17 00:00:00 2001 From: WongLaw Date: Mon, 9 Jan 2023 04:00:17 +0000 Subject: [PATCH] vits_init, test=tts --- paddlespeech/t2s/models/vits/vits.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlespeech/t2s/models/vits/vits.py b/paddlespeech/t2s/models/vits/vits.py index bc63bb995..05ec93e80 100644 --- a/paddlespeech/t2s/models/vits/vits.py +++ b/paddlespeech/t2s/models/vits/vits.py @@ -514,7 +514,7 @@ class VITS(nn.Layer): or isinstance(module, nn.Conv1DTranspose) or isinstance(module, nn.Conv2D) or isinstance(module, nn.Conv2DTranspose): - kaiming_uniform_(module.weight, a=math.sqrt(5), mode="fan_out") + kaiming_uniform_(module.weight, a=math.sqrt(5)) if module.bias is not None: fan_in, _ = _calculate_fan_in_and_fan_out(module.weight) if fan_in != 0: