如何获取静态推理模型的最后一层输出向量的形状?
收藏
我目前正在进行一个云边协同模型的评估,现在需要获取边端模型输出的向量大小。
目前只有预训练的.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
收藏
请登录后评论
paddle.summary api 可以查看参数量,输出各层的参数以及输入输出信息
参考文档:https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/summary_cn.html#summary