Keras 名称“K”未定义

Question

我正在尝试像这样加载 Keras 模型：

mpiotte_model = keras.models.load_model('./metadata/mpiotte-standard.model')

但我收到以下错误：

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-31-63426033cbe8> in <module>()
      2 steps = 0
      3 
----> 4 mpiotte_model = keras.models.load_model('./metadata/mpiotte-standard.model')
      5 model.set_weights(mpiotte_model.get_weights())

1 frames
/usr/local/lib/python3.7/dist-packages/keras/layers/core/lambda_layer.py in <lambda>(x)

NameError: Exception encountered when calling layer "lambda_3" (type Lambda).

name 'K' is not defined

Call arguments received:
  • inputs=['tf.Tensor(shape=(None, 512), dtype=float32)', 'tf.Tensor(shape=(None, 512), dtype=float32)']
  • mask=None
  • training=False

我也有这行，导入成功：

from keras import backend as K

我在 Google Colab 上运行此程序，但我不知道如何解决此问题。

Answer 1

这个导入绝对没问题：

from keras import backend as K

你可以试试这个来解决问题：

mpiotte_model = keras.models.load_model('./metadata/mpiotte-standard.model', custom_objects={"K": K})