Failed to process: (InvalidArgument) Axis should be less than 1, but received axis is 1 [Hint: Expected axis < max_dim, but received axis:1 >= max_dim:1.] [operator < elementwise_floordiv > error]

#2017 · open · 2 comments

View on GitHub ↗

DDAshley126

我自己训练了一个ser模型,使用predict_kie_token_ser.py结果好好的,但是转成serving部署后就出现Axis should be less than 1这种情况。 <img width="979" height="132" alt="Image" src="https://github.com/user-attachments/assets/5c871ab8-50fe-4069-98f0-a782e50f1ea8" /> 经检查配置没有问题,之前训练的模型也能成功部署。我尝试过重新转推理模型再转serving,所有的配置项都一致。 目前的环境: paddle-serving-app 0.9.0 paddle-serving-client 0.9.0 paddle-serving-server 0.9.0 paddlenlp 2.5.2 paddleocr 2.6 paddlepaddle 2.5.2

Comments

github-actions[bot]

Message that will be displayed on users' first issue

DDAshley126

<img width="483" height="125" alt="Image" src="https://github.com/user-attachments/assets/c1325747-26c6-47ce-8c7e-8bd3df1d5127" /> 检查后发现,paddle_serving_server.pipeline.operator.py的predict这里报错,但是feed_batch的长度都是对的上的