首页 Paddle框架 帖子详情
如何使用训练好的模型?
收藏
快速回复
Paddle框架 问答模型训练 1098 2
如何使用训练好的模型?
收藏
快速回复
Paddle框架 问答模型训练 1098 2

深度学习小白提问,希望得到大佬的详细解答。

我在运行这个项目的时候,基于U-Net卷积神经网络实现宠物图像分割-使用文档-PaddlePaddle深度学习平台,想保存我的训练模型(model)。用于下一次直接使用训练好的模型去做预测。

在末尾我加了这句:

model.save('inference_model', False)

那么,我下一次想使用这个模型的时候,应该怎么加载呢?

在另一个打算加载这个模型的py文件里,我尝试着写了这么一句:

model = paddle.Model.load("./inference_model")

但是一直报错:TypeError: load() missing 1 required positional argument: 'path'。求解答

0
收藏
回复
全部评论(2)
时间顺序
SaltyF1sh
#2 回复于2021-10
https://www.paddlepaddle.org.cn/documentation/docs/zh/tutorial/cv_case/image_segmentation/image_segmentation.html

 

0
回复
三岁
#3 回复于2021-10

可以尝试一下

paddle.load('xxxxxx.pdparams')

然后这个后面的False可以不加的

model.save('inference_model', False)
0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户