解决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牛刀小试密码爆破
Feb 03 Python
python中查找excel某一列的重复数据 剔除之后打印
Feb 10 Python
对Python新手编程过程中如何规避一些常见问题的建议
Apr 01 Python
python pandas库中DataFrame对行和列的操作实例讲解
Jun 09 Python
python中计算一个列表中连续相同的元素个数方法
Jun 29 Python
对python 多个分隔符split 的实例详解
Dec 20 Python
Python数据类型之Number数字操作实例详解
May 08 Python
Python学习笔记之自定义函数用法详解
Jun 08 Python
pytorch-神经网络拟合曲线实例
Jan 15 Python
详解Python多线程下的list
Jul 03 Python
Python + opencv对拍照得到的图片进行背景去除的实现方法
Nov 18 Python
Python机器学习实战之k-近邻算法的实现
Nov 27 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 页面编码声明方法详解(header或meta)
2010/03/12 PHP
zf框架的校验器InArray使用示例
2014/03/13 PHP
ThinkPHP 3.2 数据分页代码分享
2014/10/14 PHP
轻松实现php文件上传功能
2017/02/17 PHP
php多进程中的阻塞与非阻塞操作实例分析
2020/03/04 PHP
setInterval 和 setTimeout会产生内存溢出
2008/02/15 Javascript
javascript不同页面传值的改进版
2008/09/30 Javascript
用客户端js实现带省略号的分页
2013/04/27 Javascript
js格式化金额可选是否带千分位以及保留精度
2014/01/28 Javascript
Labelauty?jQuery单选框/复选框美化插件分享
2015/09/26 Javascript
浅谈Vue.js 组件中的v-on绑定自定义事件理解
2017/11/17 Javascript
Vue将页面导出为图片或者PDF
2020/08/17 Javascript
vue项目中极验验证的使用代码示例
2019/12/03 Javascript
jQuery实现弹出层效果
2019/12/10 jQuery
Vue使用JSEncrypt实现rsa加密及挂载方法
2020/02/07 Javascript
浅谈Vue 函数式组件的使用技巧
2020/06/16 Javascript
[01:03]DOTA2新的征程 你的脚印值得踏上
2014/08/13 DOTA
[45:38]DOTA2上海特级锦标赛主赛事日 - 1 胜者组第一轮#1Liquid VS Alliance第一局
2016/03/02 DOTA
python模拟登陆Tom邮箱示例分享
2014/01/13 Python
在Gnumeric下使用Python脚本操作表格的教程
2015/04/14 Python
Python语言生成水仙花数代码示例
2017/12/18 Python
python去掉空白行的多种实现代码
2018/03/19 Python
django反向解析和正向解析的方式
2018/06/05 Python
Python 中的range(),以及列表切片方法
2018/07/02 Python
python使用matplotlib模块绘制多条折线图、散点图
2020/04/26 Python
Python字符串的一些操作方法总结
2019/06/10 Python
获取Pytorch中间某一层权重或者特征的例子
2019/08/17 Python
python如何实现不用装饰器实现登陆器小程序
2019/12/14 Python
python几种常用功能实现代码实例
2019/12/25 Python
python实现银行实战系统
2020/02/26 Python
详解Anaconda安装tensorflow报错问题解决方法
2020/11/01 Python
小学教师自我评价
2015/03/04 职场文书
导游词之唐山景点
2019/12/18 职场文书
Go语言 go程释放操作(退出/销毁)
2021/04/30 Golang
Python使用OpenCV和K-Means聚类对毕业照进行图像分割
2021/06/11 Python
SQL模糊查询报:ORA-00909:参数个数无效问题的解决
2021/06/21 Oracle