因此,我经常运行具有不同体系结构的模型,但是具有适用于所有这些模型的代码,这些代码对已保存的模型进行推理。因此,我将在此模型的最后一层调用eval()
,如下所示:
yhat = graph.get_tensor_by_name("name_of_my_last_layer:0")
decoded_image = yhat.eval(session=sess, feed_dict={x : X})
然而,如果没有艰苦的日志解析,我不确切知道最后一层的名称是什么,我现在正在手工编写它。我已经考虑在我的图表中创建一个通用的“输出”张量,但这看起来很浪费/脆弱。有什么更好的方法?
答案 0 :(得分:0)
最好的方法是使要分析模型输出的图层或修改其名称(通过在创建图层时将name = keyword参数传递给图层函数)作为已知字符串。