【发布时间】:2020-10-23 11:12:20
【问题描述】:
我正在尝试将 TensorFlow 模型 (SavedModel) 加载到 ML.NET 并使用 ML.NET 创建一个估算器。
我正在加载模型而没有错误,但tensorFlowModel.ScoreTensorFlowModel 出现异常“System.MissingMethodException IntPtr Tensorflow.Status.op_Implicit(Tensorflow.Status)”。
我的代码:
using Microsoft.ML;
var inputNodeName = "serving_default_dense_input";
var outputNodeName = "StatefulPartitionedCall_1";
var tensorFlowModel = mlContext.Model.LoadTensorFlowModel(@"model"); //model directory contains saved_model.pb file
var pipeline = tensorFlowModel.ScoreTensorFlowModel(new[] { outputNodeName },new[] { inputNodeName }, addBatchDimensionInput: false);
已安装的 NuGet 包:
ML.NETSciSharp.TensorFlow.RedistTensorFlow.NET
这些软件包已更新到最新版本。
【问题讨论】:
-
你只是在
LoadTensorFlowModel方法中传入一个目录吗?我觉得应该是pb文件 -
@Jon
LoadTensorFlowModel方法通过。当指定了不正确的 inputNodeName 时,出现异常'在图形“grap-key-0/”中找不到操作“xyz”。所以加载模型似乎成功了。
标签: c# tensorflow ml.net