RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: “bn1.num

海洋之心

已于 2024-01-11 23:18:58 修改

阅读量951

点赞数 5

文章标签： python 开发语言人工智能分类

于 2024-01-09 12:16:17 首次发布

本文链接：https://blog.csdn.net/m0_47256162/article/details/135477315

版权

图像分类项目实战专栏收录该内容

6 篇文章 0 订阅 ¥199.90 ¥299.90

订阅专栏

超级会员免费看

在使用PyTorch加载模型时遇到RuntimeError，原因是模型结构与状态字典不匹配，特别是'num_batches_tracked'键。解决方法是通过`model.load_state_dict(state_dict, strict=False)`加载，忽略不匹配的键。但应注意，这可能导致部分层未正确初始化，理想情况是确保模型结构与状态字典一致。" 106487373,257110,Spring Boot 2.x 深入理解Specification查询,"['Spring框架', 'Java开发', '数据库操作', 'ORM框架', '微服务']

摘要由CSDN通过智能技术生成

在这里插入图片描述

问题描述：

使用 PyTorch 加载我的模型 model = load_model('checkpoint.pt') 出现如下错误：

RuntimeError Traceback (most recent call last) in () 
Load your model to this variable
model = load_model('checkpoint.pt'

了解本专栏

超级会员免费看

海洋之心

关注

5
点赞
踩
8

收藏

觉得还不错? 一键收藏
打赏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录