ImportError: cannot import name ‘get_config‘ from ‘tensorflow.python.eager.context‘ 报错解决

今天在跑高光谱分类SOTA模型SpectralNET《SpectralNET a 2D wavelet CNN for Hyperspectral Image Classification》时遇到一个问题。

报错信息：

ImportError: cannot import name 'get_config' from 'tensorflow.python.eager.context'

报错原因：Keras版本不符

模型中的Keras版本为2.4.3，我安装的时候直接
pip install Keras默认安装的是最新版的2.8.0

解决方法：

方法一：

pip install keras==xxx(模型原始的版本）

或者

pip install git+https://www.github.com/keras-team/keras-contrib.git

方法二：

pip install --upgrade tensorflow
pip install --upgrade tensorflow-gpu

我使用第一种方法，问题解决。

获取更多AI特辑：AiCharm