首页 Paddle Inference 帖子详情
如何获取静态推理模型的最后一层输出向量的形状?
收藏
快速回复
Paddle Inference 问答推理 1085 1
如何获取静态推理模型的最后一层输出向量的形状?
收藏
快速回复
Paddle Inference 问答推理 1085 1

我目前正在进行一个云边协同模型的评估,现在需要获取边端模型输出的向量大小。

目前只有预训练的.pdmodel、.pdiparams、.pdiparams.info三个文件。我使用paddle.static.load_inference_model读取它,能够得到一个Program。我不知道如何运行该Program,也不知道如何获得该Program的输出,但是知道该Program的输入形状是(None, 3, 32, 32)。

请问如何获取该模型的最后一层输出向量的形状?

附:下面是我读取该文件的方法

paddle.enable_static()
paddle_executor = paddle.static.Executor(paddle.CPUPlace())
loaded_infer_model = paddle.static.load_inference_model(infer_model_filepath, paddle_executor)
infer_program = loaded_infer_model[0]

flops = paddleslim.analysis.flops(infer_program, False)
print('FLOPs:{0}'.format(flops))

calc_output_shape(infer_program)# 还没有写的部分

    
0
收藏
回复
全部评论(1)
时间顺序
由兀C
#2 回复于2021-08

 paddle.summary api 可以查看参数量,输出各层的参数以及输入输出信息

参考文档:https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/summary_cn.html#summary

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