【TensorFlow】key xxxxx/xxxxx/Adam not found in checkpoint
在使用rmsprop或者adam优化器的时候,他们会在自己内部建立一个adam或者rmsprop变量,使用SGD优化器的时候不会新建变量,所以不会出现在这种问题。
所以在保存的时候处理一下。
这个方法好像还是不太对,再实验一下。
总之,知道原因是什么了。
vl = [v for v in tf.global_variables() if "Adam" not in v.name]
saver = tf.train.Saver(var_list=vl)
参考链接:
https://blog.csdn.net/qq_25737169/article/details/78126135
https://blog.csdn.net/qq_25737169/article/details/78125061
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!
