vits_init, test=tts

pull/2809/head
WongLaw 3 years ago
parent 56bcf13da8
commit 956a8acbc5

@ -108,7 +108,7 @@ class TextEncoder(nn.Layer):
# dist = paddle.distribution.Normal(loc=0.0, scale=attention_dim**-0.5) # dist = paddle.distribution.Normal(loc=0.0, scale=attention_dim**-0.5)
# w = dist.sample(self.emb.weight.shape) # w = dist.sample(self.emb.weight.shape)
self.emb.weight.set_value(w) # self.emb.weight.set_value(w)
self.encoder = Encoder( self.encoder = Encoder(
idim=-1, idim=-1,

Loading…
Cancel
Save