首页module 'tensorflow._api.v2.train' has no attribute 'Saver'应该怎样修改

module 'tensorflow._api.v2.train' has no attribute 'Saver'应该怎样修改

时间: 2023-07-21 21:07:15 浏览: 1108

解决tensorflow模型参数保存和加载的问题

终于找到bug原因！记一下；还是不熟悉平台的原因造成的！ Q：为什么会出现两个模型对象在同一个文件中一起运行，当直接读取他们分开运行时训练出来的模型会出错，而且总是有一个正确，一个读取错误？而直接在同一个文件又训练又重新加载模型预测不出错，而且更诡异的是此时用分文件里的对象加载模型不会出错？ model.py,里面含有 ModelV 和 ModelP，另外还有 modelP.py 和 modelV.py 分别只含有 ModelP 和 ModeV 这两个对象，先使用 modelP.py 和 modelV.py 分别训练好模型，然后再在 model.py 里加载进来： # -*- codi

在TensorFlow 2中，Saver已经被tf.compat.v1.train.Saver替换了。你可以这样修改： ```python import tensorflow.compat.v1 as tf # 创建一个tf.compat.v1.train.Saver对象 saver = tf.train.Saver() ``` 或者你也可以使用新的模型保存方法来替代Saver，例如： ```python model.save('model.h5') # 保存模型 model = tf.keras.models.load_model('model.h5') # 加载模型 ```

阅读全文