【发布时间】:2023-03-19 22:20:01
【问题描述】:
我已经训练了我的神经风格迁移模型,并在训练后获得了 .ckpt 文件。现在我想使用 tensorflow-serving 部署这个模型。我该如何继续?
【问题讨论】:
标签: tensorflow tensorflow-serving
我已经训练了我的神经风格迁移模型,并在训练后获得了 .ckpt 文件。现在我想使用 tensorflow-serving 部署这个模型。我该如何继续?
【问题讨论】:
标签: tensorflow tensorflow-serving
安装 Docker 并拉取 Docker TensorFlow 服务映像。
$docker 拉张量流/服务
将 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
【讨论】: