tensorflow ‘/biases/Adam_1’not in ckpt file

最新推荐文章于 2019-10-11 20:10:20 发布

Double_V_

最新推荐文章于 2019-10-11 20:10:20 发布

阅读量3.2k

点赞数 3

分类专栏： tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/qq_25737169/article/details/78126135

版权

tensorflow 专栏收录该内容

8 篇文章 2 订阅

订阅专栏

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

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