项目场景:
Pytorch11,
问题描述
reduction.dump(process_obj, to_child) AttributeError: Can’t pickle local object ‘InfiniteDataLoader.init..’
原因分析:
两台相同硬件配置的电脑, 一台Windows, 另一台Ubuntu,
同样的代码,在Ubuntu下运行没有问题,在Windows下运行就报了这个问题…
其中InfiniteDataLoader是对dataloader的一个wrapper类,报错最终指向了Multiprocessing,
解决方案:
试了下将data_loader中的NUM_WORKERS参数值由8改为0以后OK了。