atc模型转换报错
收藏回复举报
atc模型转换报错
t('forum.solved') 已解决
发表于2024-12-12 15:22:51
0 查看

NPU: Ascend310P3

驱动:24.1.RC3

CANN:8.0.RC3

使用atc进行模型转换,转换命令:atc --model=$PWD/models/test1.pb  --framework=3 --output=$PWD/models/out/test --soc_version=Ascend310P3 --input_format='NHWC' --input_shape="detect/input/image:1, -1,-1,3"  --dynamic_image_size="224,224;448,448"

报错

[root@localhost ocr_test]# ./atc.sh  

ATC start working now, please wait for a moment. 

... 

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

E10042: [PID: 386791] 2024-12-12-11:33:16.811.206 GenerateOfflineModel execute failed. 

        TraceBack (most recent call last): 

        op[detect/ssd_preprocessing_eval/resize_image/resize/ResizeBilinear_ascend_mbatch_batch_0], The dim of input x is not 4[FUNC:ResizeConstInferShape][FILE:image_ops.cc][LINE:1863] 

        Call InferShapeAndType for node:detect/ssd_preprocessing_eval/resize_image/resize/ResizeBilinear_ascend_mbatch_batch_0(ResizeBilinearV2) failed[FUNC:Infer][FILE:infershape_pass.cc][LINE:117] 

        process pass InferShapePass on node:detect/ssd_preprocessing_eval/resize_image/resize/ResizeBilinear_ascend_mbatch_batch_0 failed, ret:4294967295[FUNC:RunPassesOnNode][FILE:base_pass.cc][LINE:563] 

        build graph failed, graph id:0, ret:1343242270[FUNC:BuildModelWithGraphId][FILE:ge_generator.cc][LINE:1615] 

        GenerateOfflineModel execute failed.

模型由于设计问题,默认batch size为1,需要修改哪里?

我要发帖子