pytorch torch.save 模型时出现 TypeError: can't pickle _thread.RLock objects 可能造成该问题的原因

leisure2222222

于 2020-05-05 18:02:12 发布

阅读量3.9k

点赞数 2

CC 4.0 BY-SA版权

文章标签： python 深度学习机器学习人工智能

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/leisure2222222/article/details/105936221

本文探讨了在深度学习模型训练过程中使用logging模块记录信息时遇到的问题。具体是在使用logger对象并将其传递到模型中后，尝试用torch.save保存模型时出现的错误。原因是logger作为活动对象，在保存过程中无法被pickle处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

由于跑深度学习模型次数不多,坑踩的还不够多,因此今天在跑模型时出现了如题错误

其实是我第一次使用logging模块进行训练过程的一些信息进行打印和纪录

经过思考,和对比之前的训练经历,找到不同之处后查明了原因:

在main.py主函数中,我使用logger=logging.getLogger(__name_),对象然后将其传递进行了继承于 torch.nn.Module类的模型中,故在torch.save时,由于logger是活动对象,无法对其进行pickle处理保存,因此会报错而终止

评论 4

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

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