module ‘tensorflow_core.compat.v1‘ has no attribute ‘contrib‘问题的完美解决

最新推荐文章于 2025-03-31 18:56:13 发布

小柴柴是也

最新推荐文章于 2025-03-31 18:56:13 发布

阅读量3.1k

点赞数 3

分类专栏：深度学习文章标签： tensorflow 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_42115293/article/details/120296927

版权

深度学习专栏收录该内容

1 篇文章

订阅专栏

该博客主要介绍了在升级到TensorFlow2后遇到的由于`tf.contrib`模块移除导致的错误。作者指出，将`tf.contrib.rnn.BasicLSTMCell`替换为`tf.compat.v1.nn.rnn_cell.BasicLSTMCell`可以成功解决这个问题。这个更改不影响模型的功能，并且是向后兼容的解决方案。博客内容对于正在迁移TensorFlow1到TensorFlow2的开发者具有参考价值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述：

Instructions for updating:
Use keras.layers.Dense instead.
Traceback (most recent call last):
  File "run_cnn.py", line 200, in <module>
    model = TextCNN(config)
  File "D:\MY DATA\学习资料\研究生\深度学习\text-classification-cnn-rnn-master\cnn_model.py", line 43, in __init__
    self.cnn()
  File "D:\MY DATA\学习资料\研究生\深度学习\text-classification-cnn-rnn-master\cnn_model.py", line 61, in cnn
    fc = tf.contrib.layers.dropout(fc, self.keep_prob)
AttributeError: module 'tensorflow_core.compat.v1' has no attribute 'contrib'

原因：
TensorFlow 2 删除了部分v1的库
很多代码大佬们都是用v1版本写的，所以一种解决办法是将tensorflow版本降回v1，我这么干了，结果引发更大的问题，因为菜啊。。。找了好久，下面这种方法完美解决。
解决办法：
将
cell = tf.contrib.rnn.BasicLSTMCel
用
cell = tf.compat.v1.nn.rnn_cell.BasicLSTMCell
代替就可以完美的解决出现的问题。得到的结果没啥区别。