保存Tensorflow模型仅用于预测的最节省空间/内存的方法?

时间:2018-11-03 00:12:04

标签: python tensorflow

我有一个庞大的Tensorflow模型(检查点文件为4-5 GB)。我想知道除了检查点方法之外,还有没有其他方法可以保存Tensorflow模型,即空间/内存效率。

我知道一个检查点文件也可以保存所有优化程序渐变,因此也许也可以将其删除。

我的模型非常简单,只有两个嵌入矩阵,也许我只能将这些矩阵直接保存到.npy吗?

1 个答案:

答案 0 :(得分:1)

您要对检查点进行的操作是冻结。从tensorflow的官方文档中查看this page

冻结过程会从检查点剥离所有未用于正向推理的无关信息。 Tensorflow为此提供了一个易于使用的脚本,名为freeze_graph.py