Skip to content
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.

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

K2 hlg 解码问题请教 #2502

Closed
mahuichao opened this issue Apr 24, 2024 · 1 comment
Closed

K2 hlg 解码问题请教 #2502

mahuichao opened this issue Apr 24, 2024 · 1 comment
Labels

Comments

@mahuichao
Copy link

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

我查看源码(以tranformer模型为例子),在init_model.py中,在选取类的时候,并没有走k2_model,而是走的asr_model(从yaml中读取而来),如下:
image
且在对应的recognize.py中,有三个TODO,有一个对应K2的解码,所以这里很是疑惑,求解一下。
image

@Mddct
Copy link
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
Labels
Projects
None yet
Development

No branches or pull requests

2 participants