这是本文档旧的修订版!


1.paddleclas迁移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 降级后会有稍微不正常;

降级后,将可以正常导出 pdmodel+pdiparams

评论

请输入您的评论. 可以使用维基语法:
 
人工智能/paddle/paddleclas/1.paddleclas迁移timm的注意事项.1769495487.txt.gz · 最后更改: 2026/01/27 06:31