形状[-1,1]具有负尺寸

时间:2018-01-25 07:31:28

标签: python google-cloud-platform tensorflow-serving

我正在研究tensorflow上的单个类别分类问题。培训结束了,我保存了我的模型。当我尝试使用谷歌云调用预测时,我收到此错误:

{
  "error": "Prediction failed: Error during model execution: AbortionError(code=StatusCode.INVALID_ARGUMENT, details=\"Shape [-1,1] has negative dimensions\n\t [[
Node: outputs = Placeholder[_output_shapes=[[-1,1]], dtype=DT_FLOAT, shape=[?,1], _device=\"/job:localhost/replica:0/task:0/cpu:0\"]()]]\")"
}

我将输出的预测签名定义为输出变量的build_tensor_info,定义如下:

Y = tf.placeholder("float", [None, n_classes], name = 'outputs')

我的笔记本电脑和谷歌云上的培训和测试工作正常。只有当我打电话预测我才会收到此错误时才会这样做。我使用此代码来调用预测:

gcloud ml-engine predict --model ${MODEL_NAME} --json-instances request.json

我认为这是一个简单的问题,我对python / tensorflow中的形状如何工作缺失了。

以前是否还有其他人遇到此问题?

0 个答案:

没有答案