请问这里单层某一层改在cpu跑，如何实现呢？以这个 detecti...

[复制链接] · 发表于 2021-1-31 22:32:41

请问这里单层某一层改在cpu跑，如何实现呢？以这个 detectionoutput 这个 layer为的话，怎么实现，十分感谢！
3.4.Caffe 版本的 SSD 转换失败，出现 “Invalid tensor id(1),tensor(@mbox_conf_flatten_188:ut0)”的错误。

不支持 detectionoutput 这个 layer，可以删掉，改成在 CPU 做。

只看该作者 · 发表于 2021-2-1 08:39:46

输出改成这层输入前面不就好了
输出后在cpu那边单独计算这个层