我尝试构建一个tensorflow模型--在这里,我使用tf.py_func在普通python代码中创建代码的一部分。问题是,当我将模型保存到.pb文件时,.pb文件本身非常小,不包括py_func:0张量。当我试图从.pb文件加载和运行模型时,我会得到以下错误: get ValueError: callback pyfunc_0未找到。有人能帮上忙吗。
我得到了一个TensorFlow检查点和一个导出的模型,但要使用Google ML Cloud提供模型,我需要一个saved_model.pbtxt文件。看起来我需要加载检查点并使用SavedModelBuilder,但是SavedModelBuilder需要输入和输出节点名称的字典。我的问题是,给定检查点或导出的模型(如下所示),我如何通过谷歌的ML Cloud服务找到生成pbtxt文件所需的节点名称?