add chunk size and pad size in args, test=doc

pull/1634/head
TianYuan 2 years ago
parent 78219cef7b
commit 3aec266ca5

@ -93,8 +93,8 @@ def evaluate(args):
N = 0
T = 0
chunk_size = 42
pad_size = 12
chunk_size = args.chunk_size
pad_size = args.pad_size
for utt_id, sentence in sentences:
with timer() as t:
@ -109,7 +109,7 @@ def evaluate(args):
phone_ids = input_ids["phone_ids"]
else:
print("lang should in be 'zh' here!")
# merge_sentences=False here, so we only use the first item of phone_ids
# merge_sentences=True here, so we only use the first item of phone_ids
phone_ids = phone_ids[0]
with paddle.no_grad():
# acoustic model
@ -246,6 +246,11 @@ def parse_args():
type=str2bool,
default=False,
help="whether use streaming acoustic model")
parser.add_argument(
"--chunk_size", type=int, default=42, help="chunk size of am streaming")
parser.add_argument(
"--pad_size", type=int, default=12, help="pad size of am streaming")
parser.add_argument("--output_dir", type=str, help="output dir.")
args = parser.parse_args()

Loading…
Cancel
Save