-
Notifications
You must be signed in to change notification settings - Fork 4.2k
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
[BUG] slice forward crash #112
Comments
slice 的参数是 num_slice 后面接着 num_slice 个数字表示输出的各个 c,最后几个可以 -233 代表平均切分 |
slice层应该可以根据axis选择从N x C x H x W这0~3个维度中的某一个进行切分吧?目前ncnn好像只支持了按照Channel进行切分,可以加上其它维度吗? |
是我搞错了,我的caffe模型中axis=2,是对H 进行切分,而ncnn 默认只支持对channel进行切分,所以才会crash。 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
slice.cpp forward 函数中
我的proto中的参数如下
Slice slice2 1 3 to_use1 not_use2 to_use2 not_use22 3 2 8 -233
如下函数:
for (size_t i=0; i<top_blobs.size(); i++)
{
int slice = slices_ptr[i];
if (slice == -233)
{
slice = (channels - q) / (top_blobs.size() - i);
}
当输入的bottom 的 w = 11, h = 11, c =8, top_blobs.size() = 3
当 i = 2时, q=10
slice = (channels - q) / (top_blobs.size() - i);
slice = -2 ,这个明显不对
下面的 top_blob.create(w, h, slice); crash
The text was updated successfully, but these errors were encountered: