我有一个庞大的Tensorflow模型(检查点文件为4-5 GB)。我想知道除了检查点方法之外,还有没有其他方法可以保存Tensorflow模型,即空间/内存效率。
我知道一个检查点文件也可以保存所有优化程序渐变,因此也许也可以将其删除。
我的模型非常简单,只有两个嵌入矩阵,也许我只能将这些矩阵直接保存到.npy吗?
答案 0 :(得分:1)
您要对检查点进行的操作是冻结。从tensorflow的官方文档中查看this page。
冻结过程会从检查点剥离所有未用于正向推理的无关信息。 Tensorflow为此提供了一个易于使用的脚本,名为freeze_graph.py。