MobileNet-Caffe icon indicating copy to clipboard operation
MobileNet-Caffe copied to clipboard

Fail to run on GPU

Open samsun639 opened this issue 8 years ago • 3 comments

There is no problem run on CPU, but when run on GPU, it shows below error.

error_code=401; error_message=Network creation has failed. Layer conv2_1/dw.conv2_1/dw/bn.conv2_1/dw/scale: input depth must be divisible by 4 * number of groups; error_component=Dl Network; line_no=116; thread_id=-220678860

samsun639 avatar Jun 26 '17 08:06 samsun639

I am not sure what the problem is. have you changed the kernel numbers for some layers?

shicai avatar Jun 26 '17 08:06 shicai

No, I did not change anything of the prototxt. Will update if I figure out why.

Thanks.

samsun639 avatar Jun 27 '17 05:06 samsun639

OK?

ckun5 avatar Aug 02 '17 01:08 ckun5