大多数空间/内存有效的方法保存Tensorflow模型仅用于预测？

Question

我有一个巨大的Tensorflow模型(检查点文件是4-5 gbs)。我在想，除了检查点方式之外，是否还有一种不同的方法来保存Tensorflow模型，即空间/内存效率。

我知道检查点文件也保存了所有的优化器梯度，所以也许这些也可以被剪掉。

我的模型很简单，只是两个嵌入矩阵，也许我只能直接将这些矩阵保存到.npy中？

Answer 1

你想对检查点做的就是冻结它。查看tensorflow的官方文档中的此页。

冻结过程从检查点删除所有无关的信息，这些信息不用于正向推理。Tensorflow为它提供了一个易于使用的脚本，称为graph.py。