部署训练有素的模型

时间:2019-07-06 09:00:24

标签: tensorflow tensorflow-serving

我已经训练了我的神经风格转换模型,并在训练后得到了.ckpt文件。现在我想使用张量流服务来部署这个模型。我该如何进一步?

1 个答案:

答案 0 :(得分:0)

安装Docker并拉Docker Tensorflow服务映像。

$ docker pull tensorflow / serving

将SavedModel复制到容器的模型文件夹中:

$ docker cp models / serving_base:/ models /

按照https://github.com/tensorflow/serving/blob/master/tensorflow_serving/g3doc/docker.md中的说明进行操作,您应该可以运行投放图片到主机模型。

查看以下链接以了解更多详细信息- https://www.tensorflow.org/tfx/serving/docker