小编典典

TensorFlow:如何发布没有源代码的模型?

python

我正在使用Tensorflow + Python。

我很好奇是否可以在没有详细源代码的情况下发布保存的Tensorflow模型(体系结构+训练有素的变量)。我知道tf.train.Saver(),但是它看起来只保存变量,并且为了还原/运行它们,用户需要“定义”相同的体系结构。

仅出于测试/运行目的,有没有办法释放没有源代码的保存的{architecture + trained variables},以便用户可以进行查询并获得结果?


阅读 316

收藏
2021-01-20

共1个答案

小编典典

TensorFlow服务项目的目的是使这个用例简单(假设最终用户只能使用模型推断,不培训)。TensorFlow
Serving包含一个Exporter类,该类采用您tf.train.Savertf.GraphDef,用于定义您的整体模型,以及一个“签名”,用于描述模型的输入和输出。

基础教程有一个很好的介绍导出模型。

2021-01-20