_pickle.PicklingError: Can't pickle <class '__main__.aaa'>: attribute lookup aaa on __main__ failed

注:其中aaa是我的类名。
这个问题是我想保存一个自定义的类对象时,采用如下代码pickle模块 保存时出现的报错信息。

with open(f'saved_agent_{seed}.pkl', 'wb') as file:
    pickle.dump(agent, file)

出错原因

该错误通常出现在试图使用pickle模块保存自定义类对象,而这个类定义在主模块(‘main’)中。
pickle在加载对象时需要知道类的定义,但由于main模块中的类在另一个解释器中可能没有定义,所以会导致这个错误。

解决方法1

将自定义类放入一个独立的模块中,然后在主模块中导入它。这样就保证了类定义在Pickle加载时可以找到。

解决方法2

如果现在该运行的都已经运行完了,我不想再跑一遍,重新换个模块了。
那么此时就使用dill库,事实证明好像也没有什么区别,保存出来的是一个空文件。

还是乖乖重新开一个吧

posted @   芋圆院长  阅读(380)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
点击右上角即可分享
微信分享提示