nas 对搜索到的模型的中间信息进行输出
Created by: ceci3
我尝试使用paddleslim平台运行https://github.com/PaddlePaddle/PaddleSlim/blob/release/1.0.1/docs/zh_cn/tutorials/image_classification_nas_quick_start.ipynb 网址下的例程并尝试将网络模型和参数输出,但是遇到了问题,在代码中并没有将全连接层以外的其他网络层定义出,我理解其他网络层的架构是使用:archs = sanas.next_archs()[0] 总体描述的,但是当我编写输出语句时,输出语句中fluid.io.save_inference_model(dirname=save_path, feeded_var_names=[data.name], target_vars=[predict], executor=exe) 其中target_vars=[predict] predict=其他网络层+最后全连接层,也就是说:
archs = sanas.next_archs()[0]
output = fluid.layers.fc(input=output, size=10)
predict = archs+output
如果我的理解没有错误,那么输出语句应该怎么表达才能将整个网络架构的信息进行输出呢?