繁体   English   中英

convert tensorflow.keras model to keras model

[英]convert tensorflow.keras model to keras model

我有一个 EfficientNet model (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')

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

使用调查:

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 之间转换模型,我感谢我能得到的所有帮助。

由于 tensorflow 和 keras 之间的版本不兼容,您无法加载 model。

kerastensorflow升级到2.5后,您的问题将得到解决。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM