AttributeError: module ‘tensorflow‘ has no attribute ‘contrib‘解决办法

蜡笔大新home2023-11-11 21:11

在TensorFlow2 环境下执行1.X版本的代码时报错：

AttributeError: module 'tensorflow' has no attribute 'contrib'

当然第一时间想到的是利用如下代码来解决问题（大部分情况都是这样），

python 复制代码

tf.compat.v1

但是又出现以下报错

AttributeError: module 'tensorflow.compat.v1' has no attribute 'contrib'

表示这个库中也没有该函数。

查阅资料以及实验发现，是选择库的问题，需要修改如下：

python 复制代码

# 原始代码
lstm_enc = tf.contrib.rnn.LSTMCell(num_units)

# 修改后代码
lstm_enc = tf.compat.v1.nn.rnn_cell.LSTMCell(num_units)

最后运行不报错。