[英]Shape error when predicting with a trained model in tensorflow.keras
[英]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。
将keras
和tensorflow
升级到2.5
后,您的问题将得到解决。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.