Tensorflow服务-“您必须为占位符tensor\'PlaceHolder_1\'提供一个值”

Question

我使用如下所示的导出来服务我的模型：

features_placeholder = tf.placeholder(tf.float32, None)
labels_placeholder = tf.placeholder(tf.float32, None)

# Training loop code
......
# Train is finished.

# Export model
tf.saved_model.simple_save(sess,param.logs_dir + 'model_export', 
            {"features": features_placeholder}, {"binary_classif": labels_placeholder})

然后，我发出以下POST请求（原始正文）：

{“错误”：“您必须用dtype float\n\t[[节点:Placeholder_1=placeholder_output_shapes=[],dtype=dt_float,shape=,_device=\”/job:localhost/replica:0/task:0/device:cpu:0\“]]为占位符张量提供一个值\'Placeholder_1\\”

有人知道我做错了什么吗？

Answer 1

您必须确保两个占位符features_placeholder和labels_placeholder的形状与两个变量features_feed和labels_feed的形状相对应，以避免在输入字典时出现错误。