前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >tensorflow ‘/biases/Adam_1’not in ckpt file

tensorflow ‘/biases/Adam_1’not in ckpt file

作者头像
DoubleV
发布2018-09-12 15:06:40
6190
发布2018-09-12 15:06:40
举报
文章被收录于专栏:GAN&CVGAN&CV

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_25737169/article/details/78126135

在加载模型的时候,有时候会遇到这个问题,提示在ckpt文件里找不到Adam_1变量,在使用rmsprop或者adam优化器的时候,他们会在自己内部建立一个adam或者rmsprop变量,使用SGD优化器的时候不会新建变量,所以不会出现在这种问题,之前在我的博客里简要介绍了这个问题,不过和这个不太一样 http://blog.csdn.net/qq_25737169/article/details/77856961 解决这个问题可以使用下面的方法:

代码语言:javascript
复制
temp = set(tf.global_variables())
train_op = tf.train.AdamOptimizer(lr).minimize(loss, vars_list)
sess.run(tf.initialize_variables(set(tf.global_variables())-temp))

至于选择性加载网络权重的方法可以见我另一篇博客 http://blog.csdn.net/qq_25737169/article/details/78125061

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2017年09月28日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档