保存CNN权重以便重用它们的最佳方法

Question

我想通过使用Tensorflow（Python API）从头实现和训练的CNN保存权重（和偏差）。

现在我想将这些权重保存在文件中并与某人共享，以便他可以使用我的网络。但是因为我有很多重量，所以我不知道。我该怎么做？是否有建议这样做的格式？

Answer 1

Tensorflow提供了一种保存模型的方法：tensorflow.org/api_docs/python/tf/train/Saver。您的朋友也应该使用Tensorflow加载它们。您加载/保存的语言不会影响它的保存方式 - 如果您将它们保存在Python的Tensorflow中，则可以使用C ++中的Tensorflow读取它们。