在使用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.youkuaiyun.com/qq_25737169/article/details/78126135
https://blog.youkuaiyun.com/qq_25737169/article/details/78125061