Python端预测部署trt错误。
Created by: lwbmowgli
使用release0.3以及master均出现trt错误。命令如下:
python3.7 deploy/python/infer.py --model_dir=/paddle/inference_model/yolov3_darknet --image_file=/paddle/images/test.jpg --use_gpu=True --run_mode=trt_fp16
我分别尝试了paddle1.8.1和paddle2.0.0.报错信息是一样的,当我的run_mode=fluid时,不会报错。
另外,如果我自己训练了模型的时候,在导出模型的时候会报错。
python -u tools/train.py -c configs/yolov3_mobilenet_v1_fruit.yml --eval 迭代了20次。
模型保存形式如下:
报错信息如下: