ForkingPickler(file, protocol).dump(obj) TypeError: can‘t pickle Environment objects

JOYCE_Leo16

发布于 2024-03-19 17:24:41

1550

发布于 2024-03-19 17:24:41

文章被收录于专栏：计算机视觉计算机视觉

问题描述

在win系统下复现SPSR代码出现这种错误，查询资料发现是windows系统的问题。

解决方案：

因为windows操作系统的原因，在Windows中，多进程multiprocessing使用的是序列化pickle来在多进程之间转移数据，而socket对象是不能被序列化的，但是在linux操作系统上却没问题，因为在linux上多进程multiprocessing使用的是fork，所以在windows上可以改用多线程。因为网络通信属于io密集型的操作，对cpu计算要求不高，不用多进程，用多线程就行。

令num_workers=0，记得无论是参数设置还是数据集读取处理的代码，都要重新设置。

loader = torch.utils.data.DataLoader(dataset=dataset,
                                              batch_size=self.batch,
                                              shuffle=self.shuf,
                                              num_workers=0,
                                              drop_last=True)
 
parser.add_argument('--num_workers', type=int, default=0)

还有一些其他的解决方案：python3 PicklingError: Can't pickle <function <lambda> at......> attribute lookup <lambda> on __main_can't pickle <function <lambda> at 0x000001ed8215d-CSDN博客

参考：成功解决can‘t pickle Environment objects和Ran out of input_forkingpickler(file, protocol).dump(obj) typeerror-CSDN博客

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2024-03-10，如有侵权请联系 cloudcommunity@tencent.com 删除

dump