Tencent / ncnn

ncnn is a high-performance neural network inference framework optimized for the mobile platform
Other
20.38k stars 4.16k forks source link

squeezenet_ssd跑GPU结果错误 #932

Closed kelvinou closed 5 years ago

kelvinou commented 5 years ago

squeezenet_ssd跑GPU结果错误,定位到Softmax-mbox_conf_softmax层结果与CPU结果不一致,再次麻烦nihui大佬解决一下!

kelvinou commented 5 years ago

@nihui

nihui commented 5 years ago

confirmed

kelvinou commented 5 years ago

squeezenet_ssd的输出结果是正常的,但是发现原版Caffe-SSD的官方模型在CPU上结果正确,而GPU的结果却不正确(没有结果输出),同样也是定位在Softmax-mbox_conf_softmax层的输出与CPU不一致 @nihui

nihui commented 5 years ago

confirmed

nihui commented 5 years ago

squeezenet_ssd的输出结果是正常的,但是发现原版Caffe-SSD的官方模型在CPU上结果正确,而GPU的结果却不正确(没有结果输出),同样也是定位在Softmax-mbox_conf_softmax层的输出与CPU不一致 @nihui

应该都修复了 ...