解决pytorch加载模型报错TypeError: ‘collections.OrderedDict‘ object is not callable

最新推荐文章于 2024-06-02 23:52:21 发布
最新推荐文章于 2024-06-02 23:52:21 发布
阅读量372 收藏
点赞数 2
文章标签： pytorch 人工智能 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_44263065/article/details/137709596
版权

之前训练完一个模型，模型参数用代码torch.Save(model.state_dict(), 'data/model.pth')保存道model.pth文件中，用load语句加载文件时报错，出现上面的报错语句。看了一下文献，发现原因是model.pth文件中保存的是模型的参数，不能直接加载，需要先导入网络结构，再将文件中的网络结构的参数导入。

model = GRU() #导入网络结构
model.load_state_dict(torch.load('data/model.pth'))#导入网络结构参数

weixin_44263065
关注
  • 2
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
python/pytorch 个人coding中的报错/异常
驽马十驾，功在不舍；锲而不舍，金石可镂。
05-10 1万+
快速解决python中的bug，这里总有你需要的。
《Python基础教程》内容总览篇
热门推荐
weixin_43178406的博客
08-26 11万+
大家好，我是爱编程的喵喵。双985硕士毕业，现担任全栈工程师一职，热衷于将数据思维应用到工作与生活中。从事机器学习以及相关的前后端开发工作。曾在阿里云、科大讯飞、CCF等比赛获得多次Top名次。现为CSDN博客专家、人工智能领域优质创作者。喜欢通过博客创作的方式对所学的知识进行总结与归纳，不仅形成深入且独到的理解，而且能够帮助新手快速入门。个人精心开设的《Python基础课程》专栏订阅量接近900，帮助不少同学解决了Bug。
pytorch加载模型报错TypeError: ‘collections.OrderedDict‘ object is not callable
m0_47548313的博客
05-16 4734
# 导入模型 print("model loading") path_model = "./static/model/resnet_.pt" model=resnet() model = model.load_state_dict(torch.load(path_model),False) # model = model_loaded.load_state_dict(torch.load(path_model)) print("finish") 加载模型状态，首先要初始化模型状态，再加载状态 实际上，
Pytorch加载模型并进行图像分类预测
无穷酱的博客
10-20 9271
1. 整体流程 1）实例化模型 Assume that the content ofYourClass.pyis: class YourClass: # ...... If you use: from YourClassParentDir import YourClass # means YourClass from model import PythonNet net= PythonNet(T=16).eval().cuda() 2）加载模型 import ...
Pytorch：torch.optim模块
weixin_42046845的博客
11-29 518
本篇笔记主要介绍torch.optim模块，记录学习过程 在深度学习中，我们通常会使用优化算法来调整神经网络的权重和偏差，以便模型能够更好地拟合训练数据。torch.optim是PyTorch中的一个模块，它提供了各种优化算法的实现，用于自动化地优化神经网络的参数。换句话说，torch.optim可以帮助我们让模型更好地学习，从而提高性能。
pytorch源码解析1——torch.optim：优化算法接口详解
kaggle expert，全球排名前1000，清华计算机研究生，兴趣算法工程
02-16 2914
1. 优化器 Optimizer 1.0 基本用法 优化器主要是在模型训练阶段对模型可学习参数进行更新, 常用优化器有 SGD，RMSprop，Adam等 优化器初始化时传入传入模型的可学习参数，以及其他超参数如lr，momentum等 在训练过程中先调用optimizer.zero_grad()清空梯度，再调用loss.backward()反向传播，最后调用optimizer.step()更新模型参数 简单使用示例如下所示： import torch import numpy...
深入浅出Pytorch函数——torch.nn.Module
冯·诺依曼
08-19 1606
torch.nn.Module(*args, **kwargs)
将Pytorch训练的.pth模型转为tensorflow的.pb模型的一些坑与解决方法
我是托蜜的博客
01-21 3389
我用pytorch训练YOLOv4模型，训练时的log想转为tf的.pb以执行后续的工作。在转换过程中踩了不少坑，在此记录。 1. Yolo模型 网上很多转换的方法，都大同小异。但需要注意，yolo训练的log只是参数，没有网络架构，因此需要导入自己的YOLO_body，再把log的参数对应上。 import torch from nets.yolo4 import YoloBody # 我直接在yolov4的文件夹下新建的文件，所以这里直接导入了 model = YoloBody(3,3) #
解决pytorch报错:AssertionError: Invalid device id的问题
01-21
在服务器上训练的网络放到本地台式机进行infer，结果出现报错： AssertionError: Invalid device id ...以上这篇解决pytorch报错:AssertionError: Invalid device id的问题就是小编分享给大家的全部内容了，
PyTorch练手项目分享：模型微调 .docx
03-05
PyTorch练手项目分享：模型微调 .docx
Pytorch报错记录
01-20
一、TypeError: _resolve_type_from_object(): incompatible function arguments. The following argument types are supported: 1. (arg0: object, arg1: torch._C._jit_tree_views.SourceRange, arg2: Callable[...
pytorch:model.train和model.eval用法及区别详解
09-17
今天小编就为大家分享一篇pytorch:model.train和model.eval用法及区别详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
TEZ：训练pytorch模型更快rrrr......。-Python开发
05-25
tez：训练pytorch模型fastrrrr ....... tez：训练pytorch模型fastrrrr .......注意：当前，我们不接受任何拉取请求！ 所有公共关系将被关闭。 如果您需要某个功能或某些功能不起作用，请创建一个问题。 意思是“锐利...
pytorch onnx ncnn间的关系
最新发布
weixin_42605076的博客
06-02 294
PyTorch：用于模型开发和训练，提供强大的灵活性和丰富的库支持。ONNX：充当一个中间层格式，促进不同深度学习框架之间的模型互操作性和标准化。NCNN：专注于高效的模型推理，特别是在移动和嵌入式设备上。工作流程示例在 PyTorch 中开发和训练模型。将 PyTorch 模型导出为 ONNX 格式。使用 NCNN 工具将 ONNX 模型转换为 NCNN 格式。将 NCNN 模型部署到移动设备或嵌入式设备上进行高效推理。
pytorch笔记：topk
qq_40206371的博客
05-29 193
返回给定输入张量沿指定维度的最大的 k 个元素dimlargestFalsedimsortedTrue。
error: #error C++17 or later compatible compiler is required to use PyTorch.
YuQiao0303的博客
05-30 320
在安装wonder3d的时候要安装tinycudnn，
PyTorch中CrossEntropyLoss、BCELoss、BCEWithLogitsLoss的理解
u012374012的专栏
05-27 520
输出：
优化器报错TypeError: 'list' object is not callable
05-20
这个错误通常是由于在使用优化器的时候，将一个列表传递给了优化器，导致了“list对象不可调用”的错误。请检查你的代码，特别是优化器的部分，确保你传递给优化器的参数是正确的类型。可能需要使用`nn.Parameter`或`nn.ModuleList`等PyTorch提供的容器类来包装模型参数。另外，也需要检查是否有对列表进行误调用的情况。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值