tensorflow ‘/biases/Adam_1’not in ckpt file

来源:互联网 发布:c语言的string.h 编辑:程序博客网 时间:2024/05/05 18:21

在加载模型的时候,有时候会遇到这个问题,提示在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