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

update notes, test=doc

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