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