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

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

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


版权声明:本文为Sunshine_in_Moon原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。