使用张量流导出神经网络的权重

Question

我使用张量流工具编写了神经网络。一切正常，现在我想导出神经网络的最终权重以制定单一的预测方法。我怎样才能做到这一点？

Answer 1

您需要在训练结束时使用课程保存模型tf.train.Saver。

初始化对象时Saver，您需要传递要保存的所有变量的列表。最好的部分是您可以在不同的计算图中使用这些保存的变量！

Saver通过使用创建一个对象，

# Assume you want to save 2 variables `v1` and `v2`
saver = tf.train.Saver([v1, v2])

使用tf.Session对象保存变量，

saver.save(sess, 'filename');

当然，您可以添加其他详细信息，例如global_step.

您可以在以后使用该restore()函数恢复变量。恢复的变量将自动初始化为这些值。