Toybrick

标题: 请问这里单层某一层改在cpu跑，如何实现呢？以这个 detecti... [打印本页]

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

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

作者: jefferyzhang 时间: 2021-2-1 08:39
输出改成这层输入前面不就好了
输出后在cpu那边单独计算这个层

欢迎光临 Toybrick (https://t.rock-chips.com/)