将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...

我也找到了这个线程，可能会试一下，但由于efficient net有> 350层，所以它不是真正适用的How to load tf.keras models with keras

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

Answer 1

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

将keras和tensorflow升级到2.5后，您的问题将得到解决。