解决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算法学习之基数排序实例
Dec 18 Python
使用Python进行稳定可靠的文件操作详解
Dec 31 Python
对python插入数据库和生成插入sql的示例讲解
Nov 14 Python
python粘包问题及socket套接字编程详解
Jun 29 Python
python 实现批量替换文本中的某部分内容
Dec 13 Python
django 实现简单的插入视频
Apr 07 Python
Python字符串split及rsplit方法原理详解
Jun 29 Python
一些关于python 装饰器的个人理解
Aug 31 Python
Python: glob匹配文件的操作
Dec 11 Python
进行数据处理的6个 Python 代码块分享
Apr 06 Python
Python的property属性详细讲解
Apr 11 Python
python数字图像处理:图像简单滤波
Jun 28 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
如何使用PHP中的字符串函数
2006/11/24 PHP
php数组函数序列之array_unique() - 去除数组中重复的元素值
2011/10/29 PHP
使用xampp搭建运行php虚拟主机的详细步骤
2015/10/21 PHP
php mongodb操作类 带几个简单的例子
2016/08/25 PHP
thinkphp5 框架结合plupload实现图片批量上传功能示例
2020/04/04 PHP
JavaScript constructor和instanceof,JSOO中的一对欢喜冤家
2009/05/25 Javascript
jQuery中选择器小问题(新人难免遇到)
2014/03/31 Javascript
JavaScript数值转换的三种方式总结
2014/07/31 Javascript
基于JavaScript短信验证码如何实现
2016/01/24 Javascript
详解自动生成博客目录案例
2016/12/09 Javascript
JS实现淡入淡出图片效果的方法分析
2016/12/20 Javascript
获取JavaScript异步函数的返回值
2016/12/21 Javascript
bootstrap配合Masonry插件实现瀑布式布局
2017/01/18 Javascript
vuejs父子组件之间数据交互详解
2017/08/09 Javascript
JavaScript和TypeScript中的void的具体使用
2019/09/12 Javascript
LayUI switch 开关监听 获取属性值、更改状态的方法
2019/09/21 Javascript
在Vue.js中使用TypeScript的方法
2020/03/19 Javascript
json_decode 索引为数字时自动排序问题解决方法
2020/03/28 Javascript
[01:11:32]VG vs FNATIC 2019国际邀请赛小组赛 BO2 第二场 8.15
2019/08/17 DOTA
网站渗透常用Python小脚本查询同ip网站
2017/05/08 Python
Python IDLE入门简介
2017/12/08 Python
Python实现k-means算法
2018/02/23 Python
Python 3.x 安装opencv+opencv_contrib的操作方法
2018/04/02 Python
Django保护敏感信息的方法示例
2019/05/09 Python
Html5嵌入钉钉的实现示例
2020/06/04 HTML / CSS
英国家喻户晓的高街品牌:River Island
2017/11/28 全球购物
自荐信写法介绍
2014/01/25 职场文书
评析教师个人的自我评价
2014/02/19 职场文书
廉洁家庭事迹材料
2014/05/15 职场文书
领导班子四风问题对照检查材料
2014/09/27 职场文书
“四风”查摆问题自我剖析材料
2014/09/27 职场文书
导游词之镇江西津古渡
2019/11/06 职场文书
使用Selenium实现微博爬虫(预登录、展开全文、翻页)
2021/04/13 Python
《雀魂PONG☆》4月1日播出 PV角色设定情报
2022/03/20 日漫
MySQL库表太大怎么办? 数据库分库分表项目实践
2022/04/11 MySQL
如何使用注解方式实现 Redis 分布式锁
2022/07/23 Redis