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

雨点打透心脏的1/2处 2023-02-18 09:53 13阅读 0赞

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

更多python视频教程请到菜鸟教程https://www.piaodoo.com/

**错误描述：**

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() 报错处理问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多

茂名论坛https://www.hnthzk.com/

化州橘红http://www.sfkyty.com/

茂名论坛http://www.nrso.net/

源码搜藏网http://www.cntkd.net/

发表评论取消回复

表情：

评论列表（有 0 条评论，13人围观）

还没有评论，来说两句吧...

相关阅读

相关 MySQL - 报错问题解决

1、\[HY000\]\[1290\] 错误信息"\[HY000\]\[1290\] MySQL服务器使用了–secure-file-priv选项，因此无法执行此语句"表

墨蓝/ 2023年10月14日 00:15/ 0 赞/ 60 阅读

相关 TensorFlow的报错信息

学习TensorFlow的过程中最不缺的就是各种报错。痛定思痛，记录一下学习中遇到的错误~ 文章目录 1. UnicodeDecodeError

浅浅的花香味﹌/ 2023年06月01日 15:59/ 0 赞/ 30 阅读

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

解决Tensorflow2.0 tf.keras.Model.load\_weights() 报错处理问题更多python视频教程请到菜鸟教程https://www.pia

雨点打透心脏的1/2处/ 2023年02月18日 09:53/ 0 赞/ 14 阅读

相关 Tensorflow版本问题的报错收集

1、module ‘tensorflow’ has no attribute ‘set\_random\_seed’ module 'tensorflow' ha

柔情只为你懂/ 2023年02月09日 14:19/ 0 赞/ 38 阅读

相关 NoSuchMethodException 报错问题解决

首先导致这种报错的原因可能会有很多种我所遇到已知的如下所示一. 包冲突最常见导致程序不知道找哪个二.真没找到method 根据错误显示

拼搏现实的明天。/ 2022年12月26日 02:24/ 0 赞/ 280 阅读

相关【Tensorflow】Tensorflow开发报错总结

1、报错一【报错描述】在Tensorflow中引入MNIST数据集时，报如下错误： ![watermark_type_ZmFuZ3poZW5naGVpdGk_shad

傷城~/ 2022年12月08日 14:06/ 0 赞/ 360 阅读

相关解决yum报错问题

报错如下： OSError: [Errno 2] No such file or directory: '/var/cache/yum/x86_64/7/_loca

港控/mmm°/ 2022年10月16日 01:59/ 0 赞/ 438 阅读

相关 eclipse报错问题解决

1. 右键单击工程项目 ->点击 properties 2. 选择 Deployment Assembly 3. 点击 Add -> Java Build Path E

「爱情、让人受尽委屈。」/ 2022年06月16日 06:55/ 0 赞/ 324 阅读

相关解决jQuery报错问题

每次过一段时间就忘了怎么解决jquery报错问题做下记录。。。 1、选择报错的jquer文件 ![jquer文件][jquer] 2、点击右键选择 MyEclip

迷南。/ 2022年04月25日 06:50/ 0 赞/ 329 阅读

相关 tensorflow报错问题

1. tensorflow-gpu与tensorboard版本不一致 ![20190613132345160.png][] 导致运行代码时报错，错误如下 ![waterm

ゝ一纸荒年。/ 2022年01月15日 15:39/ 0 赞/ 352 阅读