python - Keras 模型到tensorflow.keras

Question

我有一个用 Keras 制作的模型(使用 Tensorflow 作为后端)。现在我想在 Tensorflow 版本中使用 Keras。

但是，替换此行

from keras.engine.topology import get_source_inputs

用这一行

from tensorflow.keras.engine.topology import get_source_inputs

不起作用。它说

ImportError: No module named 'tensorflow.keras.engine'

在 Tensorflow 中哪里可以找到 get_source_inputs 函数？

Answer 1

将 Keras 更新到版本 2.2.4 似乎对我有用!

然后，您可以通过以下方式之一导入 get_source_inputs 函数:

from keras.utils import get_source_inputs
from keras.utils.layer_utils import get_source_inputs

或者，如果您不使用 Keras 作为外部库，而是使用它的 TF 版本，那么您可以使用以下方式导入该函数:

from tensorflow.keras.utils import get_source_inputs

希望这有帮助!