PaddlePaddle / PaddleYOLO

🚀🚀🚀 YOLO series of PaddlePaddle implementation, PP-YOLOE+, RT-DETR, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOX, YOLOv5u, YOLOv7u, YOLOv6Lite, RTMDet and so on. 🚀🚀🚀
https://github.com/PaddlePaddle/PaddleYOLO
GNU General Public License v3.0
534 stars 132 forks source link

模型保存时报错 #208

Closed songh11 closed 4 months ago

songh11 commented 7 months ago

问题确认 Search before asking

Bug组件 Bug Component

No response

Bug描述 Describe the Bug

问题描述:使用 export_model.py 保存模型时报错

python3 tools/export_model.py -c=configs/yolov5/yolov5_s_300e_coco.yml --output_dir=./inference_model -o weights=weights/yolov5_s_300e_coco.pdparams

报错信息:

  File "/home/songhan/sh_share_111/Paddle/PaddleYOLO-develop/ppdet/engine/trainer.py", line 1021, in _get_infer_cfg_and_input_spec
    self.model, input_spec=input_spec, full_graph=True)
TypeError: declarative() got an unexpected keyword argument 'full_graph'

复现环境 Environment

paddleyolo版本:develop paddlepaddle版本:2.4.2

Bug描述确认 Bug description confirmation

是否愿意提交PR? Are you willing to submit a PR?

nemonameless commented 7 months ago

更新一下paddle,使用 2.5及以上版本试试

songh11 commented 7 months ago

更新一下paddle,使用 2.5及以上版本试试

感谢回复,我这边需要在 python3.6 环境跑,只能用 2.5 以下版本,我把 full_graph 去掉可以保存,请问这样保存的模型会有问题么

nemonameless commented 4 months ago

没问题的,在您的环境能跑通就行。