atc模型转换不支持torchvision.ops.nms算子吗
收藏回复举报
atc模型转换不支持torchvision.ops.nms算子吗
t('forum.solved') 已解决
发表于2023-10-11 16:17:47
0 查看

如题,在yolov8模型中加入torchvision.ops.nms来做nms,导出的onnx模型转换om模型报错

ATC start working now, please wait for a moment. 

ATC run failed, Please check the detail log, Try 'atc --help' for more information 

EZ3002: Optype [NonMaxSuppressionV7] of Ops kernel [AIcoreEngine] is unsupported. Reason: [tbe-custom]:op type NonMaxSuppressionV7 is not found in this op store.[tbe-builtin]:The op is dynamic shape, but is not configured to support dynamic shape in op store.. 

        Possible Cause: The operator type is unsupported in the operator information library due to specification mismatch. 

        Solution: Submit an issue to request for support at https://gitee.com/ascend, or remove this type of operators from your model. 

        No supported Ops kernel and engine are found for [PartitionedCall_/model.22/NonMaxSuppression_NonMaxSuppressionV6_103], optype [NonMaxSuppressionV7]. 

        Optype [StridedSliceV3] of Ops kernel [AIcoreEngine] is unsupported. Reason: [tbe-custom]:op type StridedSliceV3 is not found in this op store.[tbe-builtin]:[Dynamic shape check]: data type DT_INT64 of input [x] is not supported. All supported data type and format of tensor input0.x is: Data Type: {DT_FLOAT16,DT_FLOAT,DT_INT32,DT_UINT8,DT_BOOL,DT_INT8,DT_FLOAT16,DT_FLOAT,DT_INT32,DT_UINT8,DT_BOOL,DT_INT8}Format:{ND,ND,ND,ND,ND,ND,ND,ND,ND,ND,ND,ND}. 

        No supported Ops kernel and engine are found for [/model.22/Slice_2], optype [StridedSliceV3]. 

        build graph failed, graph id:0, ret:-1[FUNC:BuildModel][FILE:ge_generator.cc][LINE:1429] 

我要发帖子