我正在使用vgg16进行张量流转移学习。我只是在该网站上的numpy npz文件中获得的500MB权重之上训练一个小层。当我保存我的模型时,我只指定我正在训练的权重,因此模型文件很小 - 但是元文件是500MB。从阅读stackoverflow: what-is-the-tensorflow-checkpoint-meta-file起,删除元文件听起来很安全,但是我可以配置张量流来不写吗?
答案 0 :(得分:5)
您可以使用可选的tf.train.Saver.save
参数尝试write_meta_graph=False
。
https://www.tensorflow.org/versions/r0.11/api_docs/python/state_ops.html#Saver