使用使用paddle.jit.save()保存模型后,paddle.jit.load() 加载出错
收藏
使用paddle.jit.save()保存模型后,使用paddle.jit.load() 加载报错，
错误为：
ValueError: (InvalidArgument) Failed to parse program_desc from binary string.
[Hint: Expected desc_.ParseFromString(binary_str) == true, but received desc_.ParseFromString(binary_str):0 != true:1.] (at /paddle/paddle/fluid/framework/program_desc.cc:103)
车辆一品炼丹师
已解决
5#
回复于2022-03
保存时路径：/home/aistudio/model 保存成功的文件名：unet_model_000060.pdiparams unet_model_000060.pdiparams.info unet_model_000060.pdnodel 加载时的错代码：network = paddle.jit.load( '/home/aistudio/model' ) 正确代码：network = paddle.jit.load( '/home/aistudio/model/unet_model_000060' ) 原来加载的文件路径要带文件名的前缀，就这个小问题搞了大半天，吐血吐血！！！！！！！
2
收藏
请登录后评论
好像需要指定保存的输入尺寸大小
保存是保存成功了，但是就是加载不了
这个好像是保存静态图的，要不你换个函数试试
保存时路径：/home/aistudio/model
保存成功的文件名：unet_model_000060.pdiparams
unet_model_000060.pdiparams.info
unet_model_000060.pdnodel
加载时的错代码：network = paddle.jit.load( '/home/aistudio/model' )
正确代码：network = paddle.jit.load( '/home/aistudio/model/unet_model_000060' )
原来加载的文件路径要带文件名的前缀，就这个小问题搞了大半天，吐血吐血！！！！！！！
哈哈，恭喜恭喜
+1
感谢感谢，我也找了半天
感谢楼主，找了快一天了