AttributeError: module 'tensorflow._api.v2.train' has no attribute 'Saver'
时间: 2023-09-20 21:10:22 浏览: 340
AttributeError: module 'tensorflow.compat.v1' has no attribute '
这个错误通常是由于 TensorFlow 版本不兼容造成的。在 TensorFlow 2.0 之后,`tf.train.Saver` 被弃用了,而使用 `tf.saved_model.save` 和 `tf.saved_model.load` 来保存和加载模型。如果你在 TensorFlow 2.0 或更高版本中使用了 `tf.train.Saver`,就会出现上述错误。
要解决这个问题,你可以将代码中的 `tf.train.Saver` 替换为 `tf.saved_model.save` 和 `tf.saved_model.load`。如果你只是想加载一个已经训练好的模型,可以使用以下代码:
```python
import tensorflow as tf
model = tf.saved_model.load('/path/to/saved/model')
```
如果你想保存一个模型,可以使用以下代码:
```python
import tensorflow as tf
# 构建模型
model = ...
# 训练模型
...
# 保存模型
tf.saved_model.save(model, '/path/to/save/model')
```
请注意,`tf.saved_model.save` 保存的是一个包含模型权重和计算图的文件夹,你可以使用 `tf.saved_model.load` 加载它。
阅读全文