New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

K2 hlg 解码问题请教 #2502

Open

mahuichao opened this issue Apr 24, 2024 · 1 comment

mahuichao commented Apr 24, 2024

请问目前wenet-main中，对于k2 的hlg解码，也就是aishell/s0/run.sh中的第8阶段是否支持了呢。

我查看源码（以tranformer模型为例子），在init_model.py中，在选取类的时候，并没有走k2_model，而是走的asr_model（从yaml中读取而来），如下：

且在对应的recognize.py中，有三个TODO，有一个对应K2的解码，所以这里很是疑惑，求解一下。

Collaborator

Mddct commented May 6, 2024

需要自己改下，初始化model为k2 并且 model.decode 直接调用 model.hlgxxx

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment