解决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实现的Kmeans++算法实例
Apr 26 Python
Python彩色化Linux的命令行终端界面的代码实例分享
Jul 02 Python
python字典操作实例详解
Nov 16 Python
Python实现购物车程序
Apr 16 Python
Python实现string字符串连接的方法总结【8种方式】
Jul 06 Python
基于Python的ModbusTCP客户端实现详解
Jul 13 Python
Flask教程之重定向与错误处理实例分析
Aug 01 Python
python自动下载图片的方法示例
Mar 25 Python
jupyternotebook 撤销删除的操作方式
Apr 17 Python
Django-migrate报错问题解决方案
Apr 21 Python
Pytorch 高效使用GPU的操作
Jun 27 Python
pytorch 中autograd.grad()函数的用法说明
May 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
将OICQ数据转成MYSQL数据
2006/10/09 PHP
php flv视频时间获取函数
2010/06/29 PHP
ThinkPHP自动验证失败的解决方法
2011/06/09 PHP
php数组函数序列之array_search()- 按元素值返回键名
2011/11/04 PHP
PHP反射类ReflectionClass和ReflectionObject的使用方法
2013/11/13 PHP
php读取远程gzip压缩网页的方法
2014/12/29 PHP
PHP中header用法小结
2016/05/23 PHP
javascript实现获取字符串hash值
2015/05/10 Javascript
javascript实现查找数组中最大值方法汇总
2016/02/13 Javascript
Bootstrap 折叠(Collapse)插件用法实例详解
2016/06/01 Javascript
JS关闭窗口时产生的事件及用法示例
2016/08/20 Javascript
针对后台列表table拖拽比较实用的jquery拖动排序
2016/10/10 Javascript
给Easyui-Datebox设置隐藏或者不可用的解决方法
2017/05/26 Javascript
详解Vue 方法与事件处理器
2017/06/20 Javascript
react路由配置方式详解
2017/08/07 Javascript
vue实现图书管理demo详解
2017/10/17 Javascript
VueJs使用Amaze ui调整列表和内容页面
2017/11/30 Javascript
JS插入排序简单理解与实现方法分析
2019/11/25 Javascript
javascript实现贪吃蛇小游戏
2020/07/28 Javascript
[01:52]2014DOTA2西雅图邀请赛 V社开大会你不知道的小秘密
2014/07/08 DOTA
使用Python 正则匹配两个特定字符之间的字符方法
2018/12/24 Python
Python图像的增强处理操作示例【基于ImageEnhance类】
2019/01/03 Python
基于python历史天气采集的分析
2019/02/14 Python
python实现植物大战僵尸游戏实例代码
2019/06/10 Python
Python socket 套接字实现通信详解
2019/08/27 Python
Python matplotlib读取excel数据并用for循环画多个子图subplot操作
2020/07/14 Python
Python项目打包成二进制的方法
2020/12/30 Python
PyCharm+Miniconda3安装配置教程详解
2021/02/16 Python
CSS3中currentColor关键字的妙用
2016/02/27 HTML / CSS
印度化妆品购物网站:Nykaa
2018/07/22 全球购物
美国优质马术服装购买网站:Breeches.com
2019/12/16 全球购物
保护动物倡议书
2014/04/15 职场文书
工作目标责任书
2014/07/23 职场文书
先进集体申报材料
2014/12/25 职场文书
2015年学校图书室工作总结
2015/05/19 职场文书
服务器SVN搭建图文安装过程
2022/06/21 Servers