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

load_param_bin 导入模型的时候,可能会卡死在layers.resize() blobs.resze() #150

Closed
lubinBoooos opened this issue Sep 27, 2017 · 1 comment

Comments

@lubinBoooos
Copy link

当使用者偶然出现load_param_bin导入了错误的模型文件或者把模型的权值文件作为输入,这个函数中的layer_count/blob_count可能会出现负数,从而导致程序卡在layers.resize() blobs.resze()处,建议在这之前对读入的文件做一个判断,防止误操作

@nihui
Copy link
Member

nihui commented Oct 18, 2017

new param load api 检查开头特殊的 magic number

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

No branches or pull requests

2 participants