chunkformer-onnx / chunkformer_offline.md
Audrey178's picture
Upload 2 files
5bf68b0 verified
|
Raw History Blame Contribute Delete
403 Bytes

chunk_size=64 left_context_size=128 right_context_size=128

ort_session = ort.InferenceSession(onnx_path) ort_inputs = { "xs": xs.numpy(), "xs_lens": xs_lens.numpy(), "chunk_size": np.array(chunk_size, dtype=np.int64), "left_context_size": np.array(left_context_size, dtype=np.int64), "right_context_size": np.array(right_context_size, dtype=np.int64), } ort_outs = ort_session.run(None, ort_inputs)[0]