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

update notes, test=doc

上级 e0d222e6
...@@ -69,13 +69,13 @@ def ort_predict(args): ...@@ -69,13 +69,13 @@ def ort_predict(args):
voc_sess = get_sess(args, filed='voc') voc_sess = get_sess(args, filed='voc')
# am warmup # am warmup
for batch in [27, 38, 54]: for T in [27, 38, 54]:
data = np.random.randint(1, 266, size=(batch, )) data = np.random.randint(1, 266, size=(T, ))
am_sess.run(None, {"text": data}) am_sess.run(None, {"text": data})
# voc warmup # voc warmup
for batch in [227, 308, 544]: for T in [227, 308, 544]:
data = np.random.rand(batch, 80).astype("float32") data = np.random.rand(T, 80).astype("float32")
voc_sess.run(None, {"logmel": data}) voc_sess.run(None, {"logmel": data})
print("warm up done!") print("warm up done!")
...@@ -120,9 +120,7 @@ def parse_args(): ...@@ -120,9 +120,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(
......
...@@ -69,13 +69,13 @@ def ort_predict(args): ...@@ -69,13 +69,13 @@ def ort_predict(args):
voc_sess = get_sess(args, filed='voc') voc_sess = get_sess(args, filed='voc')
# am warmup # am warmup
for batch in [27, 38, 54]: for T in [27, 38, 54]:
data = np.random.randint(1, 266, size=(batch, )) data = np.random.randint(1, 266, size=(T, ))
am_sess.run(None, {"text": data}) am_sess.run(None, {"text": data})
# voc warmup # voc warmup
for batch in [227, 308, 544]: for T in [227, 308, 544]:
data = np.random.rand(batch, 80).astype("float32") data = np.random.rand(T, 80).astype("float32")
voc_sess.run(None, {"logmel": data}) voc_sess.run(None, {"logmel": data})
print("warm up done!") print("warm up done!")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册