根据Unable to deploy a Cloud ML model如果我想将我的模型部署到Google Cloud ML,我需要明确设置"input" / "output"集合,这些集合将存储对输入/输出张量的引用,如下所示:
此集合应为您的图表命名所有输入张量 . 同样,需要一个名为“outputs”的集合来命名图形的输出张量 . 假设你的图形有两个输入张量x和y,以及一个输出张量分数,这可以按如下方式完成:tf.add_to_collection(“inputs”,json.dumps({“x”:x.name,“y”:y .name}))tf.add_to_collection(“outputs”,json.dumps({“scores”:scores.name}))这里“x”,“y”和“得分”成为实际张量名称的别名(x . name,y.name和scores.name)
但是,我不知道翻译中的输入/输出张量是什么(RNN)tutorial . 没有这些知识,我无法重构代码并将我的模型部署到Google Cloud ML .
1 回答
根据下面的代码,输入是:encoder_inputs,decoder_inputs,target_weights,输出是step()返回值输出的第三个元素
https://github.com/petewarden/tensorflow_makefile/blob/master/tensorflow/models/rnn/translate/seq2seq_model.py#L170