如何与其他用户共享张量流量权重

时间:2016-12-14 11:19:00

标签: python numpy tensorflow

是否可以与其他用户共享tensorflow检查点文件(平台和CPU / GPU独立)?我分享了一个tensorflow implementation of the DeconvNet,现在我想提供训练有素的重量。我可以简单地上传保存的模型或是否有其他方式?我问,因为我读了一个教程,权重是使用{{1}}存储的,然后在重量初始化期间恢复。但这种方法用于MNIST示例,该示例使用非常小的网络。

谢谢!

2 个答案:

答案 0 :(得分:1)

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

其中一个缺点是它没有提供要提取/提取的张量的注释,因此您需要提供一些显示如何使用它的代码。

SavedModel是TensorFlow检查点格式的下一个迭代,负责处理,但它还没有太多文档。

答案 1 :(得分:0)

我使用pickle, in binary mode来转储和加载大的numpy矩阵,它运行得很好。