search

wenet-e2e / wenet

Production First and Production Ready End-to-End Speech Recognition Toolkit
https://wenet-e2e.github.io/wenet/
Apache License 2.0
4.18k stars 1.08k forks source link

MoE onnx 导出 #2646

Closed xIaott-s closed 1 week ago

xIaott-s commented 3 weeks ago

我看到历史issue中有提到如果使用了moe模块的话，无法导出onnx。但是我使用wenet/bin/export_onnx_gpu.py是可以正常导出encoder.onnx 和 decoder.onnx的，而且cer跟torch checkpoint是对齐的。想请教下是有什么坑我没注意到么？还是目前已经支持moe的onnx导出了？

xingchensong commented 1 week ago

所谓“不支持”是指没有经过验证，如果你验证没问题的话那就太好了