torch.save时报错 Can‘t pickle local object ‘_initialize.＜locals＞.ResBasicBlock.__init__.＜locals＞.lambda

最新推荐文章于 2023-03-10 17:33:27 发布
最新推荐文章于 2023-03-10 17:33:27 发布
阅读量1.4k 收藏
点赞数
文章标签： pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/SuperB666/article/details/124712666
版权

PyTorch 模型保存 pickle lambda表达式 错误修复
关键词由CSDN通过智能技术生成

今天在保存模型时遇到了如下所示的问题，查找了很多资料，没有满意的结果。

torch.save时报错 Can‘t pickle local object ‘_initialize.＜locals＞.ResBasicBlock.__init__.＜locals>.lambda

后面发现是pickle有限制，无法解析我模型文件中的lambda表达式

进入模型文件修改无法pickle的位置即可：

如图，我的是

ResBasicBlock.__init__下的lambda表达式，后面换成了函数

SuperB666
关注
torch.save 时报错Can‘t pickle local object ‘BasicBlock.__init__.＜locals＞.＜lambda＞‘
weixin_45587004的博客
04-02 1019
保存模型 虽然没有明白是什么原因，但是找到了解决方法： 将torch.save(model, path) # 直接保存整个模型方法改为torch.save(model.state_dict(), path) # 保存模型的参数 载入模型 相应的，载入模型时将model = torch.load(path) # 直接加载模型方法改为 model = Model() # 先初始化一个模型 model.load_state_dict(torch.load(path)) # 再加载模型参数 ————————————
torch.save 时报错 Can‘t pickle local object ‘_initialize.＜locals＞.patch_forward.＜locals＞.new_fwd‘
我才是一卓的博客
10-08 6074
保存模型 虽然没有明白是什么原因，但是找到了解决方法： 将torch.save(model, path) # 直接保存整个模型方法改为torch.save(model.state_dict(), path) # 保存模型的参数 载入模型 相应的，载入模型时将model = torch.load(path) # 直接加载模型方法改为 model = Model() # 先初始化一个模型 model.load_state_dict(torch.load(path)) # 再加载模型参数 ...
PYTORCH保存模型时报错AttributeError: Can‘t pickle local object ‘Stage.__init__.＜locals＞.＜lambda＞‘
m0_48095841的博客
03-10 1040
PYTORCH保存模型时报错AttributeError: Can't pickle local object 'Stage.__init__..'
pytorch保存模型出错AttributeError: Can‘t pickle local object ‘_initialize.＜locals＞.patch_forward.＜locals＞.
kewei chen
01-09 659
看了众多资料，我了解到这个问题与装饰器、多进程有关，这个问题主要发生在混合精度训练时，也可以参考一篇关于装饰器的。更换pickle框架。
【Centernet复现】AttributeError:Can‘t pickle local object ‘get_dataset.＜locals＞.Dataset
大道至简
04-28 2121
一、错误描述 AttributeError:Can't pickle local object 'get_dataset.<locals>.Dataset 二、解决办法 把涉及到numworks的值都改为0。 opts.py文件里的参数num_workers默认值改为0。 main.py文件里num_workers的值改为0。 val_loader = torch.utils.data.DataLoader( Dataset(opt, 'val'), batch_
Can‘t pickle local object ‘data.numerize.＜locals＞.＜lambda＞‘
csdn_wmk的博客
10-28 751
困扰了好久，记录一下。 报错原因是由于pickle dump 不能dump 带lambda的对象 例如代码： def numerize(self, data, user2id, item2id): user_id = map(lambda x: user2id[x], data['user_id']) item_id = map(lambda x: item2id[x], data['item_id']) pickle.dump(user_reviews, open(os
PicklingError: Can‘t pickle ＜function ＜lambda＞...attribute lookup ＜lambda＞ on __main__ failed
weixin_43407455的博客
04-25 5916
PicklingError: Can’t pickle <function at 0x000002AD125A6318>: attribute lookup on main failed 环境：win10, python3.7, jupyter notebook 报错来源：20天吃掉那只pytorch-DAY2 报错原因：cell3 + cell4 ⬅ torch.utils.data 一、cell3 + cell4 // cell3 def target_transform(t):
tensor(0.7428, grad_fn=＜DivBackward0＞) 取出数据
HJ33_的博客
07-14 5597
用模型训练计算loss的时候，loss的结果是： tensor(0.7428, grad_fn=<DivBackward0>) 如果想绘图的话，需要单独将数据取出，取出的方法是x.item() 例如： x = torch.tensor(0.8806, requires_grad=True) print(x.item()) 结果是这样的： 0.8805999755859375 不知道为什么会有数位的变化，路过的可否告知一下~ 那么在训练的时候就这样写 all_epoch_loss = []
in <module>报错_【PyTorch】torch.nn.Module 源码分析
weixin_39851809的博客
11-20 3646
torch.nn.Module 这个类的内部有多达 48 个函数，这个类是 PyTorch 中所有 neural network module 的基类，自己创建的网络模型都是这个类的子类，下边是一个示例。这篇文章就和大家一起来阅读一下这个 base class 。 import 首先是__init__和forward这两个函数。__init__中主要是初始化一些内部需要用到的state；forwa...
解决python多线程报错:AttributeError: Can't pickle local object问题
09-17
主要介绍了解决python多线程报错:AttributeError: Can't pickle local object问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
学习pickle模块，使用socket传输封装类的例子
12-10
学习pickle模块，使用socket传输封装类的例子，详情可以参考我的博客
pytorch torch.save 模型时 出现 TypeError: can't pickle _thread.RLock objects 可能造成该问题的原因
leisure2222222的博客
05-05 3729
由于跑深度学习模型次数不多,坑踩的还不够多,因此今天在跑模型时出现了如题错误 其实是我第一次使用logging模块进行训练过程的一些信息进行打印和纪录 经过思考,和对比之前的训练经历,找到不同之处后查明了原因: 在main.py主函数中,我使用logger=logging.getLogger(__name_),对象然后将其传递进行了 继承于 torch.nn.Module类的模型中,故在t...
AttributeError: Can‘t pickle local object ‘Stage.__init__.＜locals＞.＜lambda＞‘
AI浩
11-26 1158
打开serialization.py文件。
Can't pickle local object 'CDLL.__init__.<locals>._FuncPtr'
jacke121的专栏
04-19 4303
Can't pickle local object 'CDLL.__init__.<locals>._FuncPtr' 动态库代码： cv_dll=CDLL("E:\git_project\CV_client_190327\py_client\dll_call\dll\hik_client.dll") 错误原因： CDLL 不能作为参数传到多进程，必须要在进...
pytorch - swa_model模型保存的问题
菊头蝙蝠的博客
05-26 2413
在pytorch中使用swa_model 进行保存的时候会出现这样的问题。 1.首先复制一份swa_utils.py到当前目录 2.其次将AveragedModel类改成下面的样子 class AveragedModel(Module): def __init__(self, model, device=None): super(AveragedModel, self).__init__() self.module = deepcopy(model)
python 使用多线程报错：AttributeError: Can't pickle local object
火云邪神的博客
09-18 1万+
报错信息： Traceback (most recent call last): File “D:/flaskProject/test.py”, line 35, in test pool.apply(self.out, args=(i,)) File “Python37-32\lib\multiprocessing\pool.py", line 261, in apply return self...
python 多进程 AttributeError: Cant pickle local object
热门推荐
weixin_42516989的博客
06-21 2万+
AttributeError: Can’t pickle local object python闭包不支持pickle（序列化）。多进程需要函数能pickle。 有几种方式解决： 0.将函数体中的函数挪出。 这是最简单的方法。但是有时因为函数接口已经给定，需要使用函数来定义函数。考虑以下两种方法。 1.更换pickle，考虑以下代码： from pathos.multiprocessing import ProcessingPool as Pool 网址 https://github.com/uqfou
pytorch（Apex训练）保存模型报错：Can‘t pickle local object ‘_initialize.＜locals＞.patch_forward new_fwd’ 的解决方法
qq_43799400的博客
07-27 2150
在使用 Apex 进行混合精度训练时，发现经过 Apex 的 model 无法通过torch.save() 保存，报错： AttributeError: Can’t pickle local object ‘_initialize..patch_forward..new_fwd’ 我发现 经过 Apex 的 model 不能保存模型，只能保存模型参数。因此不能用 torch.save(model, ‘model.pt’) 保存模型，只能用 torch.save(model.state_dict(), ‘mo
AttributeError: Can‘t pickle local object
weixin_43564920的博客
09-22 3434
解决办法：
torch.save attributeerror: can't pickle local object 'convert_frame.<locals>
最新发布
07-16
报错"AttributeError: Can't pickle local object 'convert_frame.<locals>" 是由于torch.save()函数中调用了一个无法被pickle序列化的本地对象 'convert_frame.<locals>'。 出现这样的问题可能有几种原因。首先，...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值