Tensorflow加载预训练模型和保存模型

时间:2019-06-24 13:38:41   收藏:0   阅读:150

1. Tensorflow模型文件

(1)checkpoint
该文件是文本文件,里面记录了保存的最新的checkpoint文件以及其他checkpoint文件列表。在测试的时候,可以通过修改这个文件,指定具体使用哪个模型
(2)meta文件
这个文件保存的是计算图结构,可以理解为神经网络结构图。是一个二进制的pb格式文件,包含变量、op、集合等。
(3)ckpt文件
ckpt文件是二进制文件,保存了所有的weights、biases、gradients等变量。在tensorflow 0.11之前,保存在.ckpt文件中。0.11后,通过两个文件保存

2. 保存Tensorflow模型

参考

原文:https://www.cnblogs.com/laokanblog/p/11076270.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!