环境
paddlepaddle==2.4.0
paddlehub==2.3.1
问题描述
PaddleOCR 使用自己训练的模型时,抛出异常 'could not create a primitive descriptor for a reorder primitive',且仅在开启MKL加速时报错,不开启MKL加速则不报错。
解决方案
通过查询paddle官方issue发现,这是paddle 2.4.x版本的一个bug,升级至2.5.1可以解决该问题。
如果在升级后报错 paddle cannot import name 'RNNCell' from 'paddle.fluid.layers',则是因为2.5.0后没有fluid了,需要改一下 ’text_generation_task.py'文件引用及相关代码中对应字段。
参考:Could not create a primitive descriptor while running quantized model with MKLDNN · Issue #42972 · PaddlePaddle/Paddle · GitHub