1. 首页
  2. 编程语言
  3. Python
  4. tensorflow模型的save与restore,及checkpoint中读取变量方式

tensorflow模型的save与restore,及checkpoint中读取变量方式

上传者: 2021-05-05 01:05:30上传 PDF文件 106.80 KB 热度 14次

生成三个文件,分别是checkpoint,.ckpt.data-00000-of-00001,.ckpt.index在提取模型时,需要将模型结构再定义一遍,再将各参数加载出来这时我们想知道我在文件中到底保存了什么内容,即需要读取出checkpoint中的tensor运行后生成一个params.txt文件,在其中可以看到模型的参数。1 第一次运行后,会发现log文件夹下产生如下文件2 第二次运行后,结果如下:。本例是按照训练时间来保存的。可见程序自动载入检查点是从第15147次开始运行的。2 使用该方法,必须要定义global_step变量,否则会报错误。

下载地址
用户评论