模型保护程序元文件是巨大的,我可以配置tensorflow不写吗?

时间:2016-09-29 16:39:55

标签: tensorflow

我正在使用vgg16进行张量流转移学习。我只是在该网站上的numpy npz文件中获得的500MB权重之上训练一个小层。当我保存我的模型时,我只指定我正在训练的权重,因此模型文件很小 - 但是元文件是500MB。从阅读stackoverflow: what-is-the-tensorflow-checkpoint-meta-file起,删除元文件听起来很安全,但是我可以配置张量流来不写吗?

1 个答案:

答案 0 :(得分:5)

您可以使用可选的tf.train.Saver.save参数尝试write_meta_graph=False

https://www.tensorflow.org/versions/r0.11/api_docs/python/state_ops.html#Saver