提交 f264b912 编写于 作者: 小湉湉's avatar 小湉湉

add warmup for frontend, test=doc

上级 d592f252
...@@ -79,6 +79,13 @@ def ort_predict(args): ...@@ -79,6 +79,13 @@ def ort_predict(args):
voc_sess.run(None, {"logmel": data}) voc_sess.run(None, {"logmel": data})
print("warm up done!") print("warm up done!")
# frontend warmup
# Loading model cost 0.5+ seconds
if args.lang == 'zh':
frontend.get_input_ids("你好,欢迎使用飞桨框架进行深度学习研究!", merge_sentences=True)
else:
print("lang should in be 'zh' here!")
N = 0 N = 0
T = 0 T = 0
merge_sentences = True merge_sentences = True
...@@ -132,9 +139,7 @@ def parse_args(): ...@@ -132,9 +139,7 @@ def parse_args():
'--voc', '--voc',
type=str, type=str,
default='hifigan_csmsc', default='hifigan_csmsc',
choices=[ choices=['hifigan_csmsc', 'mb_melgan_csmsc'],
'hifigan_csmsc', 'mb_melgan_csmsc'
],
help='Choose vocoder type of tts task.') help='Choose vocoder type of tts task.')
# other # other
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册