首页 > 解决方案 > Tensorflow 'Accuracy' 对象在使用 load_model 时没有属性 '_serialized_attributes' 错误

问题描述

import tensorflow as tf
loaded_model = tf.keras.models.load_model('inception-KLDivergence')

上面的代码产生错误:

AttributeError: 'Accuracy' object has no attribute '_serialized_attributes'

模型使用model.save()格式SavedModel保存。

标签: pythontensorflowkeras

解决方案


改变你所有的from kerasfrom tensorflow.keras以便load_model()save()来自tensorflow.keras而不是keras

确保在保存模型和加载模型时都这样做

https://github.com/keras-team/keras/issues/13590#issuecomment-643889497

这为我解决了


推荐阅读