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

ncnn lstm model #41

Merged
merged 2 commits into from
Mar 14, 2020
Merged

ncnn lstm model #41

merged 2 commits into from
Mar 14, 2020

Conversation

nihui
Copy link
Contributor

@nihui nihui commented Mar 14, 2020

ncnn库需要等下面这个 PR 合并后更新下
Tencent/ncnn#1613

其中 ncnn param 手工将 Squeeze 层替换为 Reshape,并修复了后面一处 Permute 参数
因为 ncnn 不支持 batch inference,于是 LSTM 层后面的 fc 改为外层循环展开处理

#define CRNN_LSTM 0
目前暂时为0,等ncnn的.a .h更新后可以打开lstm模型

nihui added 2 commits March 14, 2020 15:14

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Copy link
Owner

@DayBreak-u DayBreak-u left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nihui 大佬 实现的ncnn crnn_lstm 推理

@DayBreak-u DayBreak-u merged commit d5fb9ef into DayBreak-u:master Mar 14, 2020
@huangchaosuper
Copy link

@nihui 请问有换工作的计划,感兴趣私信我:) wx:ok_it_is_me

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants