search

mindspore-lab / mindocr

A toolbox of OCR models, algorithms, and pipelines based on MindSpore
https://mindspore-lab.github.io/mindocr/
Apache License 2.0
174 stars 44 forks source link

在910b，cann8.0 rc1上转换ppocrv4版本特别慢 #701

Closed bltcn closed 4 weeks ago

bltcn commented 1 month ago

使用的是启智平台的 mindtorch0.3_mindspore2.3.0_torchnpu2.2.0_cann8.0镜像 +------------------------------------------------------------------------------------------------+ | npu-smi 23.0.rc2.2 Version: 23.0.0 | +---------------------------+---------------+----------------------------------------------------+ | NPU Name | Health | Power(W) Temp(C) Hugepages-Usage(page)| | Chip | Bus-Id | AICore(%) Memory-Usage(MB) HBM-Usage(MB) | +===========================+===============+====================================================+ | 2 910B | OK | 65.3 23 0 / 0 | | 0 | 0000:41:00.0 | 0 2633 / 15665 1 / 32768 | +===========================+===============+====================================================+ +---------------------------+---------------+----------------------------------------------------+ | NPU Chip | Process id | Process name | Process memory(MB) | +===========================+===============+====================================================+ | No running processes found in NPU 2 | +===========================+===============+====================================================+ 执行bash paddle2mindir.sh -m=ch_PP-OCRv4 超过半个小时第一个det的转化都还没完成

bltcn commented 1 month ago

执行日志如下： (MindSpore) [root@ze4fae7fd2404f9c8409e165921dea2f-task0-0 tools]# cat paddle2mindir.log paddle2onnx command: paddle2onnx --model_dir /tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer --model_filename inference.pdmodel --params_filename inference.pdiparams --save_file /tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer/det_db.onnx --opset_version 11 --enable_onnx_checker True >> /tmp/code/mindocr/tools/paddle2mindir.log 2>&1 [Paddle2ONNX] Start to parse PaddlePaddle model... [Paddle2ONNX] Model file path: /tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer/inference.pdmodel [Paddle2ONNX] Paramters file path: /tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer/inference.pdiparams [Paddle2ONNX] Start to parsing Paddle model... [Paddle2ONNX] Use opset_version = 11 for ONNX export. [Paddle2ONNX] PaddlePaddle model is exported as ONNX format now. 2024-06-01 05:14:26 [INFO] ===============Make PaddlePaddle Better!================ 2024-06-01 05:14:26 [INFO] A little survey: https://iwenjuan.baidu.com/?code=r8hu2s paddle2onnx Success: /tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer/det_db.onnx convert command: converter_lite --saveType=MINDIR --fmk=ONNX --optimize=ascend_oriented --modelFile=/tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer/det_db.onnx --outputFile=/tmp/code/mindocr/tools/ppocr_models/models/ch_PP-OCRv4_det_infer/det_db_dynamic_output --configFile=dynamic_config.txt >> /tmp/code/mindocr/tools/paddle2mindir.log 2>&1 [2024-06-01-05:14:39.947.384]9672 WARNING: Option input_shape_range is deprecated and will be removed in future version,please use input_shape instead 可以看到，到转换步就死掉了

bltcn commented 4 weeks ago

可以正常转换完成