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')
你就当是一个刚认识的绅士闹了个笑话吧

浙公网安备 33010602011771号