我可以保存带有其值的图表而不保存输入吗？

Question

我的网络权重由手动tf.assign填充，现在我想用权重值保存网络，但没有占位符输入。似乎tf.train.Saver只有在feed_dict可用时才有效，tf.train.export_meta_graph仅保存网络结构。我尝试了pickle和dill，但他们都有错误。这种储蓄有没有更好的解决方案？

Answer 1

占位符将输入数据转换为张量，所以我猜它们是图表的重要部分，我不明白为什么你不想包含它们。

即使您使用tf.assign，也可以冻结图形，这意味着将结构与权重相结合。冻结的做法是将Tensorflow变量转换为常量。

您必须保存图表的结构：

gdef = g.as_graph_def()
tf.train.write_graph(gdef,".","graph.pb",False)

然后保存重量（训练后）

saver.save(sess, 'tmp/my-weights')

根据https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/lite

中的教程冻结图表

之后，您可以使用图表。