解决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字符串连接方式汇总
Aug 21 Python
Python实现基于权重的随机数2种方法
Apr 28 Python
python任务调度实例分析
May 19 Python
自动化Nginx服务器的反向代理的配置方法
Jun 28 Python
python 异或加密字符串的实例
Oct 14 Python
python读取word文档,插入mysql数据库的示例代码
Nov 07 Python
Python 闭包,函数分隔作用域,nonlocal声明非局部变量操作示例
Oct 14 Python
浅谈对pytroch中torch.autograd.backward的思考
Dec 27 Python
python七种方法判断字符串是否包含子串
Aug 18 Python
Python3中小括号()、中括号[]、花括号{}的区别详解
Nov 15 Python
Python基于mediainfo批量重命名图片文件
Dec 29 Python
python3操作redis实现List列表实例
Aug 04 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判断输入不超过mysql的varchar字段的长度范围
2011/06/24 PHP
无需数据库在线投票调查php代码
2016/07/20 PHP
Laravel框架控制器的middleware中间件用法分析
2019/09/30 PHP
laravel 错误处理,接口错误返回json代码
2019/10/25 PHP
判断ie的两种简单方法
2013/08/12 Javascript
jQuery Validate 验证,校验规则写在控件中的具体实例
2014/02/27 Javascript
jQuery将所有被选中的checkbox某个属性值连接成字符串的方法
2015/01/24 Javascript
jQuery中$(function() {});问题详解
2015/08/10 Javascript
JS中Eval解析JSON字符串的一个小问题
2016/02/21 Javascript
js 获取经纬度的实现方法
2016/06/20 Javascript
jQuery基础_入门必看知识点
2016/07/04 Javascript
js中的面向对象入门
2017/03/06 Javascript
vue.js实现用户评论、登录、注册、及修改信息功能
2020/05/30 Javascript
详解React 16 中的异常处理
2017/07/28 Javascript
Node中使用ES6语法的基础教程
2018/01/05 Javascript
AngularJS中的作用域实例分析
2018/05/16 Javascript
js中apply()和call()的区别与用法实例分析
2018/08/14 Javascript
解决vue跨域axios异步通信问题
2019/04/17 Javascript
浅谈layui 数据表格前后台传值的问题
2019/09/12 Javascript
学习python (1)
2006/10/31 Python
Python的垃圾回收机制深入分析
2014/07/16 Python
Python实现简单的可逆加密程序实例
2015/03/05 Python
Python3简单实例计算同花的概率代码
2017/12/06 Python
Python中模块(Module)和包(Package)的区别详解
2019/08/07 Python
Python3 解决读取中文文件txt编码的问题
2019/12/20 Python
Python django框架开发发布会签到系统(web开发)
2020/02/12 Python
Python3 读取Word文件方式
2020/02/13 Python
python 爬虫如何实现百度翻译
2020/11/16 Python
python3判断IP地址的方法
2021/03/04 Python
html5 div布局与table布局详解
2016/11/16 HTML / CSS
Bench加拿大官方网站:英国城市服装品牌
2017/11/03 全球购物
党员思想汇报范文
2013/12/30 职场文书
高一历史教学反思
2014/01/13 职场文书
物业管理委托协议(2篇)
2014/09/23 职场文书
感恩母亲节活动总结
2015/02/10 职场文书
Nginx反向代理至go-fastdfs案例讲解
2021/08/02 Servers