Caffe2填坑系列(10)----编译成功,在python中使用时报”key already registered. Offending key: ImageInput“

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Sunshine_in_Moon/article/details/90384296

原因:ImageInput这个Op被注册了多次,我在编写自己的.cc文件时,是以image_input.cc为模板,前面我都改成了自己Op的名字,包括REGISTER_CPU_OPERATOR(),OPERATOR_SCHEMA().唯独没有改NO_GRADIENT(),造成了上述错误。

具体代码可参考core/register.h

展开阅读全文

没有更多推荐了,返回首页