当前位置: 首页 > news >正文

pytorch加载模型出现错误

大概的错误长下面这样:

问题出现的原因:

​很明显,我就是犯了第一种错误。

网上的修改方法:

我觉得按道理哈,确实,蓝色部分应该是可以把问题解决了的​。​但是我没有解决,因为我犯了另外一个错误。(大家看了后面可能会理解我为啥这样说,因为我另起了一个文件但我没有注意到这个问题)关于上图标的红色的那个解决方案,完全不建议使用,很拉跨​!不要用​!治标不治本,就是用来糊弄的​!

其实我的错误就是原因2:

我在把我修改的模型训练完了之后,用了原来的的模型​进行加载。其实单说大家可能不会理解,因为我改了一下午才悟出来,哎​。。。

​用代码来理解一下吧:

model1是原本模型的,这个模型训练完成后会生成一个pth1,同理,model2训练完会生成一个pth2。

在训练阶段呢,这两个文件都是可以正常训练的,这时候,如果你不对你的测试脚本做任何更改,那么当你用你的测试脚本去测试你训练的模型效果是否好时,只有pth1能正常加载进去,而pth2​或许大概率是会出错的。

​那么如何更改呢:

我们先看一下这一部分的代码:

这几行的代码就是在加载你的pth​。我们可以看到在第21行的那个Network,这个Network指的是你pth1的Network,所以当你在加载你的pth2时,会出现参数不匹配的情况!so......

我们只需将pth2的​Network导进来:

ok。​

http://www.lryc.cn/news/345087.html

相关文章:

  • 如何在Mac上恢复格式化硬盘的数据?
  • 华为OD机试 - 手机App防沉迷系统(Java 2024 C卷 100分)
  • 搜维尔科技:光学动作捕捉系统用于城市公共安全智慧感知实验室
  • 保研面试408复习 4——操作系统、计网
  • 实战攻防中关于文档的妙用
  • 【使用ChatGPT的API之前】OpenAI API提供的可用模型
  • 【C语言】模拟实现深入了解:字符串函数
  • 钩子函数onMounted定义了太多访问MySQL的操作 导致数据库异常
  • Excel文件解析---超大Excel文件读写
  • TypeScript基础:类型系统介绍
  • 【Unity】Unity项目转抖音小游戏(一) 项目转换
  • element-ui 中修改loading加载样式
  • QT登录界面,(页面的切换)
  • 计算机毕业设计 | vue+springboot汽车销售管理系统(附源码)
  • 一款开源的原神工具箱,专为现代化 Windows 平台设计,旨在改善桌面端玩家的游戏体验
  • python日常消费数据占比分析总结年消费方向
  • MySQL变量的浮点数问题处理
  • MWeb Pro for Mac:功能强大的Markdown博客编辑器
  • 基于FPGA实现的HDMI TO MIPI扩展显示器方案
  • 2024年美国市场亚太游戏品牌数字广告洞察报告
  • DDD面试题:DDD聚合和表的对应关系是什么 ?(来自蚂蚁面试)
  • 【华为】路由策略小实验
  • docker安装elasticsearch:7.17.21
  • 10.Java对象内置结构
  • 【ITK配准】第十五期 基于运动算法的可变形配准样例
  • CSP-j 计算机硬件
  • Java中使用RediSearch进行高效数据检索
  • NASA数据集——全球土壤顶部 1 厘米土壤湿度的网格估算值25km分辨率
  • 翼支付——风控场景中图模型的范式变迁
  • Edge视频增强功能