如何命名要在Tensorflow服务中使用的Tensorflow模型?

时间:2017-04-03 19:39:46

标签: tensorflow tensorflow-serving

执行tensorflow_model_server二进制文件时,它需要一个模型名称命令行参数model_name

如何在训练期间指定模型的名称,以便在运行tensorflow_model_server时指定它?

示例:bazel-bin/tensorflow_serving/model_servers/tensorflow_model_server --port=9000 --model_name=model --model_base_path=../models/model/version/

2 个答案:

答案 0 :(得分:1)

如果我错了,请纠正我,但看起来您实际上可以使用命令行参数model_name为模型命名,并且它不必匹配tensorflow图中的任何内容。

从源代码: https://github.com/tensorflow/serving/blob/master/tensorflow_serving/model_servers/main.cc#L41 To specify model name (default "default"): --model_name=my_name

答案 1 :(得分:1)

是的,型号名称仅用于客户端向服务器发出请求时的标识。

我正在测试Tensorflow服务分配模型名称:

--model_name="My_Model_Name"