如何与其他用户共享 tensorflow 权重

Question

是否可以与其他用户共享 tensorflow 检查点文件（平台和 CPU/GPU 独立）？我分享了一个tensorflow implementation of the DeconvNet，现在我想提供经过训练的权重。我可以简单地上传保存的模型还是有另一种 tf 方式？我问是因为我阅读了一个教程，权重是使用numpy.savetxt 存储的，然后在权重初始化期间恢复。但是这种方法被用于使用非常小的网络的 MNIST 示例..

谢谢！

Answer 1

您可以保存元图 + 提供代码来恢复和运行您的模型 -- http://tensorflow.org/how_tos/meta_graph

这样做的一个缺点是它不提供要馈送/获取的张量的注释，因此您需要提供一些代码来展示如何使用它。

SavedModel 是 TensorFlow 检查点格式的下一个迭代，它可以解决这个问题，但它还没有太多文档。

Answer 2

我使用pickle, in binary mode 来转储和加载大型 numpy 矩阵，效果很好。