tensorflow2.0保存layers.Activation无法序列化问题

在tensorflow2.0保存layers.Activation层为计算图模型

报错:ValueError: ('Cannot serialize', <tensorflow.python.framework.function._OverloadedFunction object at 0x000000CAD7716E80>)
无法对Activation进行序列化

解决方式:升级到tensorflow2.2版本。
pip install -upgrade tensorflow

保存模型方式:

(1)保存为h5文件形式,h5文件中包含模型架构和权重

mymodel.save('efficientdet.h5')

重构模型
reconstructed_model = tf.keras.models.load_model('efficientdet.h5')

(2)直接保存为SavedModel形式

mymodel.save('efficientdet', save_format='tf)
reconstructed_model = tf.keras.models.load_model('efficientdet')

(3)利用tf.saved_model.save()保存为SavedModel形式

tf.saved_model.save(my_model, 'saved_model_files/1')
tf.saved_model.load('saved_model_files/1')

posted @ 2020-05-30 16:27  平手友梨奈ii  阅读(414)  评论(0)    收藏  举报