mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Save meta data to exported ONNX models (#968)

Browse Source
This commit is contained in:
Fangjun Kuang 2023-03-27 11:39:29 +08:00 committed by GitHub
parent 7155769c19
commit 8c3ea93fc8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/librispeech/ASR/pruned_transducer_stateless3/export-onnx.py
View File

@ -273,6 +273,16 @@ def export_encoder_model_onnx(
}, },
) )
meta_data = {
"model_type": "conformer",
"version": "1",
"model_author": "k2-fsa",
"comment": "stateless3",
}
logging.info(f"meta_data: {meta_data}")
add_meta_data(filename=encoder_filename, meta_data=meta_data)
def export_decoder_model_onnx( def export_decoder_model_onnx(
decoder_model: OnnxDecoder, decoder_model: OnnxDecoder,

10
egs/librispeech/ASR/pruned_transducer_stateless5/export-onnx.py
View File

@ -296,6 +296,16 @@ def export_encoder_model_onnx(
}, },
) )
meta_data = {
"model_type": "conformer",
"version": "1",
"model_author": "k2-fsa",
"comment": "stateless5",
}
logging.info(f"meta_data: {meta_data}")
add_meta_data(filename=encoder_filename, meta_data=meta_data)
def export_decoder_model_onnx( def export_decoder_model_onnx(
decoder_model: OnnxDecoder, decoder_model: OnnxDecoder,

10
egs/librispeech/ASR/pruned_transducer_stateless7/export-onnx.py
View File

@ -291,6 +291,16 @@ def export_encoder_model_onnx(
}, },
) )
meta_data = {
"model_type": "zipformer",
"version": "1",
"model_author": "k2-fsa",
"comment": "stateless7",
}
logging.info(f"meta_data: {meta_data}")
add_meta_data(filename=encoder_filename, meta_data=meta_data)
def export_decoder_model_onnx( def export_decoder_model_onnx(
decoder_model: OnnxDecoder, decoder_model: OnnxDecoder,