从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

希望这能有所帮助！