【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


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部