Toybrick

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

rbk

注册会员

积分
69
发表于 2021-1-31 22:32:41    查看: 8134|回复: 1 | [复制链接]    打印 | 显示全部楼层
请问这里单层某一层改在cpu跑,如何实现呢?以这个 detectionoutput 这个 layer为的话,怎么实现,十分感谢!
3.4.Caffe 版 本 的 SSD 转 换 失 败 , 出 现 “Invalid tensor id(1),tensor(@mbox_conf_flatten_188:ut0)”的错误。

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

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2021-2-1 08:39:46 | 显示全部楼层
输出改成这层输入前面不就好了
输出后在cpu那边单独计算这个层
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表