PaddlePaddle / PaddleClas

A treasure chest for visual classification and recognition powered by PaddlePaddle
Apache License 2.0
5.32k stars 1.15k forks source link

怎么训练出不带reshape2算子的general_PPLCNetV2_base模型? #3142

Closed lemon3853 closed 1 month ago

lemon3853 commented 1 month ago

训练步骤:

  1. 使用配置文件进行训练ppcls/configs/GeneralRecognitionV2/GeneralRecognitionV2_PPLCNetV2_base.yaml
  2. 导出推理模型 python tools/export_model.py
  3. 使用Paddle-Liteopt工具统计模型算子信息发现带有arm架构不支持的reshape2算子 image
  4. 使用Paddle-Liteopt工具统计shitu_android_demo自带模型算子信息 https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/lite/ppshituv2_lite_models_v1.0.tar image

    1. PaddleClas版本以及PaddlePaddle版本:PaddlePaddle release/2.5.2 PaddleClas develop
    2. 涉及的其他产品使用的版本号:Paddle-Lite develop
    3. 训练环境信息: a. 具体操作系统,Linux b. Python版本号,如Python3.8 c. CUDA/cuDNN版本, 如CUDA11.7
    4. 完整的代码(相比于repo中代码,有改动的地方)、详细的错误信息及相关log
cuicheng01 commented 1 month ago

这个主要是导出的问题,你可以尝试使用其他paddle版本导出试试

lemon3853 commented 1 month ago

这个主要是导出的问题,你可以尝试使用其他paddle版本导出试试

已解决,降级到PaddlePaddle 2.4版本,谢谢大佬