AttributeError: module 'tensorflow.compat.v1.compat.v1' has no attribute 'contrib'
时间: 2023-06-22 07:28:36 浏览: 194
python中利用xml.dom模块解析xml的方法教程
`AttributeError: module 'tensorflow.compat.v1.compat.v1' has no attribute 'contrib'` 的错误提示表明在 TensorFlow 2.x 中找不到 `contrib` 模块。在 TensorFlow 2.x 中,`contrib` 模块已经被删除。
如果您需要使用 TensorFlow 1.x 中的 `contrib` 模块,可以考虑使用 TensorFlow 2.x 中的 `tensorflow.compat.v1` 模块来实现。具体来说,您可以将代码中的 `tf.contrib.xxx` 替换为 `tf.compat.v1.contrib.xxx`,如下所示:
```
import tensorflow.compat.v1 as tf
...
init_op, init_feed = tf.train.init_from_checkpoint(checkpoint_dir, assignment_map=weight_dict)
```
其中,`checkpoint_dir` 表示检查点文件所在的目录,`weight_dict` 表示要初始化的变量字典。
另外,建议在 TensorFlow 2.x 中尽可能地使用 Keras 接口来搭建神经网络模型,因为 Keras 接口更加简单易用,而且与 TensorFlow 2.x 完全兼容。
阅读全文