部署Tensorflow模型,避免使用Tensorflow服务

时间:2017-05-09 11:26:54

标签: deployment tensorflow

有没有好的方法来保存然后运行张量流模型而不使用Tensorflow服务。我的模型很简单,只需要顺序服务就可以在我的电脑上进行推理。所以我认为应该有一种更好的方法来避免使用过于丰富和复杂的Tensorflow服务系统。

我现在正在做的是在训练后保存所有重量和训练课程;并且,为了推断,我再次定义图形,恢复会话和加载权重。这是一种合理的方式吗?

1 个答案:

答案 0 :(得分:1)

您可以使用Flask(http://flask.pocoo.org/)。在您定义的其中一条路线中,您可以查询模型。这对你有帮助吗?