背景:
在本地可以运行,但是服务器上显示如下AttributeError: module 'tensorflow.keras.layers' has no attribute 'MultiHeadAttention'
(服务器上只安装了tensorflow-gpu没有安装tensorflow)
解决方法:
参照
tf.keras.layers.MultiHeadAttention | TensorFlow v2.12.0
在服务器上pip install tensorflow==2.6.0,并将keras版本改为2.6.0
问题解决