onnx_infer.sh 690 Bytes
Newer Older
sunzhq2's avatar
sunzhq2 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# python simple_asr_pipeline.py \
#     --model-dir /home/sunzhq/workspace/yidong-infer/conformer/onnx_models_batch24/transformer_lm \
#     --output-dir ./results \
#     --aishell-data-dir /data/datasets/0 \
#     --device gpu
#     # --batch_size 24


python simple_asr_pipeline.py \
    --model-dir /home/sunzhq/workspace/yidong-infer/conformer/onnx_models_batch24/transformer_lm \
    --output-dir ./results \
    --aishell-data-dir /data/datasets/0 \
    --device gpu \
    --batch-size 24

# python simple_asr_pipeline.py --model-dir ./models --skip-data-prep


# from simple_asr_pipeline import run_single_file_inference
# results = run_single_file_inference('./models', 'test.wav')