使用 tensorflow 将图形和权重保存在文件中

Question

我正在尝试使用 tensorflow 上的训练模型保存文件，我正在尝试使用这个简单的图表：

import tensorflow as tf

a = tf.placeholder(tf.float32)
b = tf.placeholder(tf.float32)
adder_node = a + b  # + provides a shortcut for tf.add(a, b).

builder = tf.saved_model_builder.SavedModelBuilder(".")

sess = tf.Session()
builder.add_meta_graph([tag_constants.SERVING])
builder.save()

print(sess.run(adder_node, {a: 3, b: 4.5}))
print(sess.run(adder_node, {a: [1, 3], b: [2, 4]}))

但它给了我一个错误：

AttributeError: 'module' object has no attribute 'saved_model_builder'

说它在tensorflow模块上不存在，我按照这个官方教程：https://www.tensorflow.org/programmers_guide/saved_model

如何将训练好的模型保存在 tensorflow 中？

Answer 1

教程没更新，把tf.saved_model_builder.SavedModelBuilder改成tf.saved_model.builder.SavedModelBuilder。