首页 > 其他 > 详细

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

时间:2020-05-30 17:52:35      阅读:71      评论:0      收藏:0      [点我收藏+]

在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‘)

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

原文:https://www.cnblogs.com/techi/p/12993371.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!