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

发布时间 2023-12-11 20:50:56作者: 芋圆院长

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

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

出错原因

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

解决方法1

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

解决方法2

如果现在该运行的都已经运行完了,我不想再跑一遍,重新换个模块了。
那么此时就使用dill库