Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
export_asr_onnx.sh 894 Bytes
Newer Older
sunzhq2's avatar
update conformer  
sunzhq2 committed
1
2
3
4
5
6
7
8
9
10
11
12
13 
# python export_asr_onnx.py \
#     --asr_model_file exp/asr_train_asr_conformer3_raw_char_batch_bins4000000_accum_grad4_sp/valid.acc.ave_10best.pth \
#     --asr_config exp/asr_train_asr_conformer3_raw_char_batch_bins4000000_accum_grad4_sp/config.yaml \
#     --output_encoder ./onnx_models/asr_encoder.onnx \
#     --output_decoder ./onnx_models/asr_decoder.onnx \
#     --stats_file exp/asr_stats_raw_sp/train/feats_stats.npz

python export_asr_onnx.py \
    --asr_model_file exp/asr_train_asr_conformer3_raw_char_batch_bins4000000_accum_grad4_sp/valid.acc.ave_10best.pth \
    --asr_config exp/asr_train_asr_conformer3_raw_char_batch_bins4000000_accum_grad4_sp/config.yaml \
    --output_encoder ./onnx_models/asr_encoder.onnx \
    --output_decoder ./onnx_models/asr_decoder.onnx
    # --stats_file exp/asr_stats_raw_sp/train/feats_stats.npz # Removed from script, add back if needed later