gpt4 book ai didi

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

转载 作者:qq735679552 更新时间:2022-09-27 22:32:09 28 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章浅谈tf.train.Saver()与tf.train.import_meta_graph的要点由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

(一)、tf.train.Saver()

(1). tf.train.Saver() 是用来保存tensorflow训练模型的,默认保存全部参数 。

(2). 用来加载参数,注:只加载存储在data中的权重和偏置项等需要训练的参数,其他一律不加载,包括meta文件中的图也不加载 。

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

(二)、tf.train.import_meta_graph

(1). 用来加载meta文件中的图,以及图上定义的结点参数包括权重偏置项等需要训练的参数,也包括训练过程生成的中间参数,所有参数都是通过graph调用接口get_tensor_by_name(name="训练时的参数名称")来获取 。

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

(三)、总结

(1). 保存使用tf.train.Saver() 。

(2). 加载可以使用tf.train.import_meta_graph(".meta文件"),直接通过训练参数名称就可以获取需要的参数,但需要提前知道训练时的参数名称才能获取,要懂得tensorflow命名规则 。

(3). tf.train.Saver("./checkpoints目录/")加载的缺点是只加载了训练参数,并且必须定义与之相同(shape,dtype要相同,tf.type要相同,如:我是placeholder,你也是tf.placeholder)方能使用,当你要获取训练中间参数时,需要和训练过程一样搭建相同的网络. 。

补充:tf.train.import_meta_graph报KeyError 。

我在模型恢复时,在执行tf.train.import_meta_graph的时候报错 。

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

后来发现,我的模型是在服务器上训练的,服务器上tensorflow版本是1.11.0,而我在本地电脑上执行的 tf.train.import_meta_graph,我本地的tensorflow是1.5.0,我将tensorflow更新到1.11.0后,就解决了.

以上为个人经验,希望能给大家一个参考,也希望大家多多支持我.

原文链接:https://blog.csdn.net/sinat_36618660/article/details/98665482 。

最后此篇关于浅谈tf.train.Saver()与tf.train.import_meta_graph的要点的文章就讲到这里了,如果你想了解更多关于浅谈tf.train.Saver()与tf.train.import_meta_graph的要点的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com