Error(s) in loading state_dict for ModelMain

最新推荐文章于 2024-09-30 11:03:09 发布
最新推荐文章于 2024-09-30 11:03:09 发布
阅读量5.7k 收藏 1
点赞数 2
分类专栏： pytorch知识宝典
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jacke121/article/details/81332031
版权
在尝试加载ModelMain的state_dict时遇到错误。问题出现在使用nn.DataParallel和cuda()时，移除这些代码会导致报错。
摘要由CSDN通过智能技术生成

 

原代码：

net = ModelMain(config, is_training=is_training)

net.train(is_training)

net = nn.DataParallel(net)

net = net.cuda()

Error(s) in loading state_dict for ModelMain

 

原因：

cuda版的torch，把代码去掉了就会报错：

net = nn.DataParallel(net)

net = net.cuda()

 
AI算法网奇
关注
专栏目录
订阅专栏
RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: “bn1.num
CSDN 精品推荐
01-09 1113
在这个特定的情况下，错误信息中提到了多个“num_batches_tracked”键。这类键在某些 PyTorch 版本中存在，而在其他版本中可能不存在。方法会加载所有匹配的键，而忽略任何不匹配的键，从而避免了这个错误。但是，这种方法应谨慎使用，因为它可能掩盖模型结构和状态字典之间的不一致问题。这样做会使得 PyTorch 忽略不匹配的键，只加载那些与模型结构相匹配的键。这通常发生在模型的结构和要加载的状态字典不完全匹配时。解决这个问题的一个常见方法是在加载状态字典之前，明确地忽略不匹配的键。
torch RuntimeError: Error(s) in loading state_dict for CRNN:
hzhj的博客
01-13 3470
加载训练的模型报如下错误。 >>> model.load_state_dict(torch.load(model_path)) Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/usr/local/lib/python2.7/dist-pa...
解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题
09-16
主要介绍了解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
多GPU训练的模型，转换加载到单GPU测试或预训练
bank777777的博客
08-02 1876
多GPU训练的模型，转换加载到单GPU测试或预训练 报错：RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1. 加载的模型是3个GPU进行训练的，而你的计算机只有1个GPU state_dict = model.state_dict() weights_path = torch.load(opt.weights_path)['model_sta
解决报错RuntimeError: Error(s) in loading state_dict for InpaintGenerator: Missing key(s) in state_dict
最新发布
长路漫漫亦灿灿
09-30 455
报错提示是因为在加载模型的state_dict时，模型中的一些关键参数（比如"conv_
pytorch使用多个GPU训练及单个GPU训练的相关问题
qq_44289607的博客
03-08 6574
1 pytorch使用多个GPU同时训练 在pytorch上使用多个GPU(在同一台设备上，并非分布式）进行训练是件非常容易的事情，只要在源代码中添加（修改）两行代码即可。 把模型放在GPU上： device = torch.device("cuda:0") model.to(device) 将tensor复制到GPU上 mytensor = my_tensor.to(device) 注意：调用my_tensor.to(device)会在GPU上返回一个新的my_tensor副本，而不是重写
pytorch 多GPU训练的模型在单卡或CPU上加载
kk的博客
06-21 4045
多卡训练，单卡加载部署。举例：在多GPU上并行训练，在单GPU或CPU上加载。 这种情况要防止参数保存的时候没有加module,那么保存的参数名称是module.conv1.weight，而单卡的参数名称是conv1.weight，这时就会报错，找不到相应的字典的错误。 此时可以通过手动的方式删减掉模型中前几位的名称，然后重新加载。 ...
PyTorch多GPU训练模型——使用单GPU或CPU进行推理的方法
qq_44949041的博客
09-07 3991
使用DataParallel`和DistributedDataParallel进行多GPU并行训练并报错模型，在推理阶段采用单个GPU或者CPU加载模型。加载时容易出现两种错误：IndexError: list index out of range 和 Missing key(s) in state_dict:。这时可以采用map_location=device等方法进行解决。
如何把多GPU模型用单GPU的方式加载
weixin_68494476的博客
07-31 269
【代码】如何把多GPU模型用单GPU的方式加载。
RuntimeError: Error(s) in loading state_dict for CE_Net_: Missing key(s) in state_dict: “MobileNet“
weixin_45561795的博客
11-14 2602
Pytorch预测时，加载权重遇到这种问题： RuntimeError: Error(s) in loading state_dict for CE_Net_: Missing key(s) in state_dict: “MobileNetV3_encoder.conv0.0.weight”, “MobileNetV3_encoder.conv0.1.weight”, “MobileNetV3_encoder.conv0.1.bias”, “MobileNetV3_encoder.conv0.1.runn
RuntimeError: Error(s) in loading state_dict for SSD
wlw8023diya的博客
07-30 1735
error: (pytorch) ubuntu-gpu@UbuntuGPU:~/wangmiao/ssd-pytorch-master$ python get_dr_txt.py Loading weights into state dict... Traceback (most recent call last): File "get_dr_txt.py", line 131, in <module> ssd = mAP_SSD() File "/home/ubuntu-
Pytorch——使用单/多个GPU训练模型
要相信光
02-25 2260
文章目录1.设置CUDA_VISIBLE_DEVICES2.定义device，并将模型移植到GPU上3.在训练里，传入模型数据移植到GPU上代码示例服务器上实际训练效果： 1.设置CUDA_VISIBLE_DEVICES # 若使用服务器多卡训练 import os # 指定对程序可见的GPU编号 # 表示只有第1,2,3块GPU可见，其他GPU不可用，并且第1块GPU编号已经变成第0块 os.e...
pytorch加载多GPU模型和单GPU模型
热门推荐
CV_YOU的博客
01-27 1万+
有时候,我们用pytorch进行多卡GPUs训练时候,保存模型应该用下面语句: torch.save(model.module.state_dict(), model_out_path) 但是忘记加module了,直接用 torch.save(model.state_dict(), model_out_path) 所以加载模型会遇到模型中参数名字多了module的关键字而报错,所以用下面加载模...
多GPU--简单运行，多卡模型加载保存
混吃等死，研究生在读，毕业无望，偶尔搞点好玩的
02-26 2510
学习记录
35李沐动手学深度学习v2/多GPU训练，数据并行，简洁实现
求道问术
09-21 354
李沐动手学深度学习v2/多GPU训练，数据并行，简洁实现
【pytorch】多GPU训练模型的保存和加载
rocking_struggling的博客
08-09 1497
pytorch中多GPU模型加载和保存需要注意的一些问题
【已解决】代码报错RuntimeError: Error(s) in loading state_dict for BertModel: Missing key(s) in state_dict:
一个手掰橙的博客
08-25 1648
load_state_dict方法参数的官方说明 strict 参数默认是true，他的含义是 是否严格要求state_dict中的键与该模块的键返回的键匹配。
RuntimeError: Error(s) in loading state_dict for BertClassifier 模型不匹配
fghjbjhgb的博客
08-15 4299
RuntimeError: Error(s) in loading state_dict for BertClassifier: size mismatch for lstm.weight_ih_l0: copying a param with shape torch.Size([400, 768]) from checkpoint, the shape in current model is torch.Size([360, 768]). size mismatch for lstm.weight_h
【pytorch】深度学习单卡训练升级到多卡训练（nn.DataParallel原理），以手写数据集分类为例
种一棵树最好的时间是十年前，其次是现在
03-22 1752
然后，nn.DataParallel会将输入数据划分为多个小批次，每个小批次分配到不同的GPU上，并使用torch.nn.parallel.scatter函数将输入数据复制到各个GPU上。接下来，nn.DataParallel会在每个GPU上分别计算模型的输出，并使用torch.nn.parallel.gather函数将各个GPU上的输出合并起来，得到最终的输出。最后，nn.DataParallel会将每个GPU上的输出合并起来，得到最终的输出，并计算损失函数和梯度。
runtimeerror error in loading state_dict for get_model
07-27
- *1* *2* *3* [PyTorch加载模型时报错RuntimeError: Error(s) in loading state_dict for *****: Missing key(s) in state...](https://blog.csdn.net/mj412828668/article/details/130014232)[target="_blank" ...
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI算法网奇

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值