如何使用训练好的模型?
收藏
深度学习小白提问,希望得到大佬的详细解答。
我在运行这个项目的时候,基于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
收藏
请登录后评论
可以尝试一下
然后这个后面的False可以不加的