modified some vars name
This commit is contained in:
parent
d08779d61e
commit
54bd759625
|
@ -1,5 +1,5 @@
|
||||||
# Fastspeech
|
# Fastspeech
|
||||||
Paddle fluid implementation of Fastspeech, a feed-forward network based on Transformer. The implementation is based on [FastSpeech: Fast, Robust and Controllable Text to Speech](https://arxiv.org/abs/1905.09263).
|
PaddlePaddle fluid implementation of Fastspeech, a feed-forward network based on Transformer. The implementation is based on [FastSpeech: Fast, Robust and Controllable Text to Speech](https://arxiv.org/abs/1905.09263).
|
||||||
|
|
||||||
## Dataset
|
## Dataset
|
||||||
|
|
||||||
|
|
|
@ -67,12 +67,12 @@ def main(args):
|
||||||
|
|
||||||
with dg.guard(place):
|
with dg.guard(place):
|
||||||
with fluid.unique_name.guard():
|
with fluid.unique_name.guard():
|
||||||
transformerTTS = TransformerTTS(cfg)
|
transformer_tts = TransformerTTS(cfg)
|
||||||
model_dict, _ = load_checkpoint(
|
model_dict, _ = load_checkpoint(
|
||||||
str(args.transformer_step),
|
str(args.transformer_step),
|
||||||
os.path.join(args.transtts_path, "transformer"))
|
os.path.join(args.transtts_path, "transformer"))
|
||||||
transformerTTS.set_dict(model_dict)
|
transformer_tts.set_dict(model_dict)
|
||||||
transformerTTS.eval()
|
transformer_tts.eval()
|
||||||
|
|
||||||
model = FastSpeech(cfg)
|
model = FastSpeech(cfg)
|
||||||
model.train()
|
model.train()
|
||||||
|
@ -105,7 +105,7 @@ def main(args):
|
||||||
mel_lens, enc_slf_mask, enc_query_mask, dec_slf_mask,
|
mel_lens, enc_slf_mask, enc_query_mask, dec_slf_mask,
|
||||||
enc_dec_mask, dec_query_slf_mask, dec_query_mask) = data
|
enc_dec_mask, dec_query_slf_mask, dec_query_mask) = data
|
||||||
|
|
||||||
_, _, attn_probs, _, _, _ = transformerTTS(
|
_, _, attn_probs, _, _, _ = transformer_tts(
|
||||||
character,
|
character,
|
||||||
mel_input,
|
mel_input,
|
||||||
pos_text,
|
pos_text,
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
# TransformerTTS
|
# TransformerTTS
|
||||||
PaddlePaddle fluid implementation of TransformerTTS, a neural TTS with Transformer. The implementation is based on [Neural Speech Synthesis with Transformer Network](https://arxiv.org/abs/1809.08895).
|
PaddlePaddle fluid implementation of TransformerTTS, a neural TTS model with Transformer. The implementation is based on [Neural Speech Synthesis with Transformer Network](https://arxiv.org/abs/1809.08895).
|
||||||
|
|
||||||
## Dataset
|
## Dataset
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue