MobileNet-Caffe
MobileNet-Caffe copied to clipboard
Fail to run on GPU
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
I am not sure what the problem is. have you changed the kernel numbers for some layers?
No, I did not change anything of the prototxt. Will update if I figure out why.
Thanks.
OK?