这是本文档旧的修订版!
1.paddle迁移timm的注意事项
paddlepaddle 版本选择问题
首先是paddlepaddle的版本选择问题,因为我是5090,没得选了,只能用
python -m pip install paddlepaddle-gpu==3.3.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu130/
但是 paddleclas看起来适配更好的是 paddlepaddle的2.6.x 版本,所以会有一些坑。
例如:导出的推理模型无法加载,因为使用 paddlepaddle3 导出的模型是 json+pdiparams 模式; paddleclas需要的是 pdmodel+pdiparams 模式;
需要配置环境变量:FLAGS_enable_pir_api=0; 强迫 paddlepaddle3 进行一定的降级;
而且目前发现,谨慎选择使用 paddle.matmul 和 paddle.einsum , paddlepaddle3 降级后会有稍微不正常;
评论