将 tensorflow.keras 模型转换为 keras 模型

Question

我有一个 EfficientNet 模型 (tensorflow.keras==2.4)，想使用 innvestigate 检查结果，但它需要 keras==2.2.4

培训代码：

tensorflow.keras.__version__ # 2.4

model = tf.keras.applications.EfficientNetB1(**params)

# do training

model.save('testModel')

我已将模型保存为文件，但无法将其加载到 Keras 2.2.4。这就是我卡住的地方，我不知道如何转换模型。

使用调查：

keras.__version__ # 2.2.4

keras.model.load_model('testModel')  # Error

# some more stuff...

我也找到了这个线程，可能会尝试，但由于高效网络具有 > 350 层，它并不真正适用 How to load tf.keras models with keras

我不知道是否真的可以在 tensorflow.keras 和 keras 之间转换模型，感谢我能得到的所有帮助。

Answer 1

由于 tensorflow 与 keras 之间的版本不兼容，您无法加载模型。

一旦您将keras 和tensorflow 升级到2.5，您的问题就会得到解决。