解决Tensorflow2.0 tf.keras.Model.load_weights() 报错处理问题


Posted in Python onJune 12, 2020

错误描述:

1、保存模型:model.save_weights('./model.h5')

2、脚本重启

3、加载模型:model.load_weights('./model.h5')

4、模型报错:ValueError: You are trying to load a weight file containing 12 layers into a model with 0 layers.

问题分析:

模型创建后还没有编译,一般是在模型加载前调用model.build(input_shape), 但我通过Dataset将输入已经变为dict格式了,暂时没找这样输入怎么匹配input_shape参数

解决方法:

model.fit(train_dataset, epochs=0)

将epochs设为0,这样模型在编译的同时不会训练数据,减少耗费的时间,之后就可以正常加载保存的参数了

补充知识:调用Kears中kears.model.load_model方法遇到的问题和解决方法

之前一直使用tf和pytorch,就算是tf也是tf.estimator用得比较多,很少使用keras,最近尝试使用kears快速训练和部署一些分类任务,在使用load_model的时候遇到一些问题

问题1:

SystemError: unknown opcode

原因是因为模型定义用到了lambda

gap = Lambda(lambda x: x[0]/x[1], name = 'RescaleGAP')([gap_features, gap_mask])

我在python3.5的环境训练的模型,python3.6的环境load模型。两个环境的lambda有差异,这个问题。

问题2:

ValueError: Unknown metric function:****

我的错误是

ValueError: Unknown metric function:top_2_accuracy

因为在构建模型时,使用了自己定义的top_2_accuracy方法,所以在load_model时需要将top_2_accuracy做为参数传进去

from keras.models import load_model
from keras.metrics import top_k_categorical_accuracy
 
def top_2_accuracy(in_gt, in_pred):
  return top_k_categorical_accuracy(in_gt, in_pred, k=2)
 
model = load_model("model.h5",custom_objects={'top_2_accuracy': top_2_accuracy})

以上这篇解决Tensorflow2.0 tf.keras.Model.load_weights() 报错处理问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python查询Mysql时返回字典结构的代码
Jun 18 Python
python服务器端收发请求的实现代码
Sep 29 Python
python实现微信接口(itchat)详细介绍
Oct 23 Python
python使用tensorflow深度学习识别验证码
Apr 03 Python
Python把csv数据写入list和字典类型的变量脚本方法
Jun 15 Python
wxPython实现整点报时
Nov 18 Python
在python里创建一个任务(Task)实例
Apr 25 Python
Pycharm中如何关掉python console
Oct 27 Python
Python图像读写方法对比
Nov 16 Python
python实现定时发送邮件
Dec 23 Python
Python3爬虫ChromeDriver的安装实例
Feb 06 Python
python绘图模块之利用turtle画图
Feb 12 Python
python + selenium 刷B站播放量的实例代码
Jun 12 #Python
解决Keras自带数据集与预训练model下载太慢问题
Jun 12 #Python
keras导入weights方式
Jun 12 #Python
keras读取h5文件load_weights、load代码操作
Jun 12 #Python
Python matplotlib 绘制双Y轴曲线图的示例代码
Jun 12 #Python
keras的siamese(孪生网络)实现案例
Jun 12 #Python
基于python实现模拟数据结构模型
Jun 12 #Python
You might like
Windows下IIS6/Apache2.2.4+MySQL5.2+PHP5.2.1安装配置方法
2007/05/03 PHP
php计算当前程序执行时间示例
2014/04/24 PHP
PHP mkdir()无写权限的问题解决方法
2014/06/19 PHP
关于PHP开发的9条建议
2015/07/27 PHP
Smarty高级应用之缓存操作技巧分析
2016/05/14 PHP
laravel框架 api自定义全局异常处理方法
2019/10/11 PHP
jQuery 选择器理解
2010/03/16 Javascript
JavaScript在for循环中绑定事件解决事件参数不同的情况
2014/01/20 Javascript
jquery实现在光标位置插入内容的方法
2015/02/05 Javascript
javascript基础知识分享之类与函数化
2016/02/13 Javascript
JS实现的颜色实时渐变效果完整实例
2016/03/25 Javascript
深入理解javascript作用域第二篇之词法作用域和动态作用域
2016/07/24 Javascript
浅谈AngularJs指令之scope属性详解
2016/10/24 Javascript
Bootstrap3 模态框使用实例
2017/02/22 Javascript
JavaScript实现弹出广告功能
2017/03/30 Javascript
Vue2.0如何发布项目实战
2017/07/27 Javascript
js求数组中全部数字可拼接出的最大整数示例代码
2017/08/25 Javascript
最新Javascript程序员面试试题和解题方法
2017/11/23 Javascript
《javascript少儿编程》location术语总结
2018/05/27 Javascript
JS简单数组排序操作示例【sort方法】
2019/05/17 Javascript
Vue使用轮询定时发送请求代码
2020/08/10 Javascript
[01:16:28]DOTA2-DPC中国联赛 正赛 iG vs Magma BO3 第二场 2月23日
2021/03/11 DOTA
python八大排序算法速度实例对比
2017/12/06 Python
Python装饰器用法实例总结
2018/02/07 Python
详解python中sort排序使用
2019/03/23 Python
Python中的list与tuple集合区别解析
2019/10/12 Python
python用类实现文章敏感词的过滤方法示例
2019/10/27 Python
Tensorflow实现部分参数梯度更新操作
2020/01/23 Python
利用python中集合的唯一性实现去重
2020/02/11 Python
python argparse模块通过后台传递参数实例
2020/04/20 Python
使用pyplot.matshow()函数添加绘图标题
2020/06/16 Python
Python3.9新特性详解
2020/10/10 Python
瑞贝卡·泰勒官方网站:Rebecca Taylor
2016/09/24 全球购物
lookfantastic荷兰:在线购买奢华护肤、护发和化妆品
2018/11/27 全球购物
大学生个人简历自我评价
2013/11/16 职场文书
教你用Python matplotlib库制作简单的动画
2021/06/11 Python