首页 Paddle框架 帖子详情
请问paddle静态图如何取出中间层结果
收藏
快速回复
Paddle框架 问答深度学习模型训练 1300 3
请问paddle静态图如何取出中间层结果
收藏
快速回复
Paddle框架 问答深度学习模型训练 1300 3

在实用paddle的过程中,算法得到一个inferenced_model(实用fluid.save_inference_model)给到工程团队,工程团队使用load_inference_model得到program,feed和fetch_list,由于debug需要,需要取出program中所有layer的output结果,但是尝试了各种方式都不奏效。请问怎么才能得到所有层的output结果呢?

2
收藏
回复
全部评论(3)
时间顺序
A
AIStudio207440
#2 回复于2021-05

您好,请问您的问题解决了吗?我也想提取静态图网络层间结果,但不知道怎么做,求指教啊~

0
回复
阳光的2小米
#3 回复于2021-06
inference_program, feed_target_names, fetch_targets = fluid.io.load_inference_model(dirname=path, executor=exe, model_filename='inference.pdmodel',params_filename='inference.pdiparams')
out_name = 'linear_1.tmp_1'
test = inference_program.block(0).var(out_name)
new_program = inference_program._prune_with_input(feeded_var_names=['inputs'], targets=[test])
res = exe.run(new_program, feed={'inputs':img}, fetch_list=[test])
print(np.shape(res))

试试这段代码

0
回复
l
lawfay
#4 回复于2021-06

可以试试拿到所有想看的out的name,然后fetch出来

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