Export the model to ONNX issue

[duhumail]

Recently I tried to export model to onnx , but encountered a prompt：

2024-08-31 16:52:35,222 INFO [checkpoint.Py:112] Loading checkpoint from k2fsa-zipformer-chinese-english-mixed/exp/epoch-99.Pt Traceback (most recent call last): File"/home/project/icefall/icefall/egs/librispeech/ASR/./Pruned_transducer_stateless7_streaming/export-onnx.Py", line 672, in main() File"/root/miniconda3/envs/icefall/lib/python3.10/Site-packages/torch/autograd/grad_mode.Py", line 27, in decorate_context return func(*args, **kwargs) File"/home/project/icefall/icefall/egs/librispeech/ASR/./Pruned_transducer_stateless7_streaming/export-onnx.Py", line 520, in main load_checkpoint(f"{params.Exp_dir}/epoch-{params.Epoch}.Pt", model) File"/home/project/icefall/icefall/icefall/checkpoint.Py", line 126, in load_checkpoint model.Load_state_dict(checkpoint["model"], strict=strict) File"/root/miniconda3/envs/icefall/lib/python3.10/Site-packages/torch/nn/modules/module.Py", line 1604, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\N\t{}'.Format( RuntimeError: Error(s) in loading state_dict for Transducer:

my model is from https://huggingface.co/csukuangfj/k2fsa-zipformer-chinese-english-mixed
the export command is
./pruned_transducer_stateless7_streaming/export-onnx.py
--tokens k2fsa-zipformer-chinese-english-mixed/data/lang_char_bpe/tokens.txt
--use-averaged-model 0
--epoch 99
--avg 1
--decode-chunk-len 32
--exp-dir k2fsa-zipformer-chinese-english-mixed/exp/

Has anyone encountered a similar problem? I used the official example to export ncnn or onnx and everything was normal, but other models seemed to have similar problems. Is there any matching problem between the export command and the script?

thanks a lot for any help.

[csukuangfj]

Would you mind posting all of the logs from the terminal?